将 Conv1D 层从 pytorch 转换为 tensorflow/keras

Question

I want to create an equal keras layer from this source:

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

My Input is shaped (Batch_size,Channel,Width) This Layer is compiled to:

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

By pytorch. How can I express this Layer in tensorflow? I have so far this:

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

But I am not confident that this will is the right approach.

Answer 1

In tensorflow's keras you write something like:

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