如何将 Conv1d 转换为 Conv2d？ [火炬]答案

【问题标题】：How to convert Conv1d into Conv2d? [PyTorch]如何将 Conv1d 转换为 Conv2d？ [火炬]
【发布时间】：2020-04-06 01:39:57
【问题描述】：

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

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

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

【问题讨论】：

【解决方案1】：

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

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

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

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

【讨论】：