如何将 Conv1d 转换为 Conv2d？ [PyTorch]

Question

PyTorch是否可以使用Conv2d来执行Conv1d ？ 这个问题可能看起来很奇怪，但我需要使用与conv1d不兼容的工具，但它适用于conv2d 。

如果我有Conv1d(in,out, kernel_size=3, stride=stride, padding=1, bias=False)怎么办？ unsqueeze 可以帮助我吗？

我对AvgPool1d (-> AvgPool2d) and MaxPool1d (-> MaxPool2d)有同样的问题。

Answer 1

Conv2D 主要是 Conv1D 的通用版本。 您当然可以使用 Conv2D 的退化版本来重现一维卷积 - 您需要向数据添加另一个维度：

data_pnt = data_pnt [..., numpy.newaxis]

您还需要指定内核大小 - 您将选择一维内核 - 例如：

Conv2d(in,out, kernel_size=(3,1), <Other Parameters>)