如何使用 nn.conv2d 做与 nn.conv1d 一样的 function？

Question

我正在努力在 FPGA 上部署 python CNN，我遇到了一个问题。

我有一个使用torch.conv1d层 function 的 1d CNN，我正在使用的软件不支持它： vitis-ai 1.3 。

我正在弄清楚是否有某种方法可以使用nn.conv2d来代替nn.conv1d做同样的工作？

例如：

Conv1d(1, 32, 3, bias = True) => Conv2d(... 

有没有办法做到这一点？ （不会损失太多性能？）

谢谢你。

Answer 1

给定一维输入，内核大小(n,1)的二维卷积等效于内核大小(n,)的一维卷积：

nn.Conv1d(in_channels, out_channels, n)

# for 1d input, is equivalent to:
nn.Conv2d(in_channels, out_channels, (n,1))

---

请注意，您必须unsqueeze(-1)输入以使其具有正确的 2d 卷积形状，但您可以通过应用 Squeeze squeeze(-1)将 output 恢复为原始形状：

F.conv1d(inputs, filters)

# is equivalent to:
F.conv2d(inputs.unsqueeze(-1), filters.unsqueeze(-1)).squeeze(-1)