将 Conv1D 层从 pytorch 转换为 tensorflow/keras

Question

我想从此来源创建一个相等的 keras 层：

Layer=torch.nn.Conv1d(in_features, out_features, 1)

我的输入形状为 (Batch_size,Channel,Width) 此图层编译为：

Conv1d(10, 256, kernel_size=(1,), stride=(1,))

通过 pytorch。 如何在 tensorflow 中表达这一层？ 到目前为止，我有这个：

layer1 = tf.keras.layers.Conv1D(in_features-out_features+1, kernel_size=1)

但我不相信这将是正确的做法。

Answer 1

在 tensorflow 的 keras 中，您可以编写如下内容：

layer1 = tf.keras.layers.Conv1D(filters=256, kernel_size=1)(layer0)