怎么把Conv1d转换成Conv2d？ [PyTorch]

Question

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

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

AvgPool1d (-> AvgPool2d) and MaxPool1d (-> MaxPool2d)也有相同的问题。

Answer 1

Conv2D主要是Conv1D的通用版本。当然，您可以使用简并版本的Conv2D来再现1D卷积- 您需要在数据中添加另一个维度：

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

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

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