在Pytorch中的2d网格上应用2d卷积

Question

我正在尝试做一个二维卷积，一个二维网格，由下面形状的张量表示

[batch_dim, width, height]

其中，第一维度是批次维度，第二和第三维度表示2d网格

当我尝试使用以下代码应用2d卷积时：

nn.Conv2d(1, 32, kernel_size=3, stride=1)

我得到以下错误：

RuntimeError: Expected 4-dimensional input for  
4-dimensional weight 32 1 1 1, but got 3-dimensional
input of size [1, 132, 7] instead

为什么对于具有一个通道的2d网格，它需要一个4d张量？我应该如何在2d网格上实现2d卷积？

Answer 1

它应该是batch size, channel, height, width格式的，所以是1通道的2d网格。使用一个通道不会改变任何事情，不要自己实现它，使用pytorch实现。