将 Conv1D 层从 pytorch 转换为 tensorflow/keras答案

【问题标题】：Translating Conv1D Layer from pytorch to tensorflow/keras将 Conv1D 层从 pytorch 转换为 tensorflow/keras
【发布时间】：2021-03-19 08:09:29
【问题描述】：

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

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

我的输入已成形（Batch_size,Channel,Width）该层编译为：

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

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

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

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

【问题讨论】：

标签： python tensorflow machine-learning keras pytorch

【解决方案1】：

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

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

【讨论】：

所以过滤器参数就是out_features？
是的，就是layer1的filter个数（通道号）