Tensorflow 2.2，Lambda 層中的 tf.nn.conv1d

Question

我想在 Lambda 層中執行卷積，但我無法讓它以任何方式工作。

    kernel = [1.0,2.0,1.0]  # weighted moving average
    x = [   # history_size=5, num_features=10
      [1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0],
      [2.0,2.0,2.0,2.0,2.0,2.0,2.0,2.0,2.0,2.0],
      [3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0],
      [4.0,4.0,4.0,4.0,4.0,4.0,4.0,4.0,4.0,4.0],
      [5.0,5.0,5.0,5.0,5.0,5.0,5.0,5.0,5.0,5.0],
    ]
    k = tf.constant(kernel, dtype=tf.float32)
    y = tf.nn.conv1d(x, k, stride=1, padding='SAME')

我意識到在上面的例子中尺寸不正確，但這是我數據的實際格式。 訓練樣本的形狀為(history_size, num_features)並且內核必須沿着 history_size 分別對每個特征進行卷積。 任何幫助，將不勝感激。 我找不到有關如何手動執行 tf.nn.conv1d 的示例。

Answer 1

您可以numpy.convolve()使用numpy.convolve() 。

import numpy as np

kernel = [1.0,2.0,1.0]  # weighted moving average
x = [   # history_size=5, num_features=10
  [1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0],
  [2.0,2.0,2.0,2.0,2.0,2.0,2.0,2.0,2.0,2.0],
  [3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0,3.0],
  [4.0,4.0,4.0,4.0,4.0,4.0,4.0,4.0,4.0,4.0],
  [5.0,5.0,5.0,5.0,5.0,5.0,5.0,5.0,5.0,5.0],
]

output = []
for i in range(len(x)):
    output.append(list(np.convolve(x[i], kernel, mode = 'same')))
output

'''
[[3.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 3.0],
 [6.0, 8.0, 8.0, 8.0, 8.0, 8.0, 8.0, 8.0, 8.0, 6.0],
 [9.0, 12.0, 12.0, 12.0, 12.0, 12.0, 12.0, 12.0, 12.0, 9.0],
 [12.0, 16.0, 16.0, 16.0, 16.0, 16.0, 16.0, 16.0, 16.0, 12.0],
 [15.0, 20.0, 20.0, 20.0, 20.0, 20.0, 20.0, 20.0, 20.0, 15.0]]
'''

您可以嘗試根據文檔更改最適合您的mode 。