我想从以下代码创建一个等效的Keras层:

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

我的输入形状为(批次大小,通道,宽度)。此层在PyTorch中编译为:

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

如何在TensorFlow中表达这个层？我目前的尝试是:

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

但我对这种方法是否正确没有信心。

回答：

在TensorFlow的Keras中，你可以这样写:

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