如何创建过滤器不跨一维移动的转换层

How to create a conv layer where the filter doesn't move across one dimension

换句话说,我希望过滤器在一个维度上覆盖图像,并且只在另一个方向上移动。

如果我有一个形状为 (h, w) 的输入(忽略批量大小和输入通道数),我希望过滤器的形状为 (x, w),其中 x<h。创建形状为 (x, w) 的过滤器并使用 'VALID' 的填充是否是正确的做法(VALID 的想法将强制过滤器仅在过滤器 [= =22=] 在图像的宽度维度上)?

是否还有其他更好的方法?

是的,你可以那样做。另一种是使用 stride 参数指定每个维度中的步幅。您可以将其他维度的步幅设置为 0。我认为 stride=(1, 0) 是正确的语法。