RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 192]

今天跑代码的时候遇到了这个错误： RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 192] ,at /opt/conda/conda-bld/pytorch_1524

在这里插入图片描述

调试之后发现是如下有问题： 源代码在这里： 这是class里init中相应的部分

在这里插入图片描述

def forward的相应的问题

在这里插入图片描述

问题就在于y1 = self.fc_1(y)这里，fc_1也是同fc一样的全连接层，而全连接层输入的尺寸只有两维！ 从图上可以看到，y输出的尺寸是[2，576，1，1] 在y1 = self.fc_1(y)的时候，因为y是四维的，所以会自动变成两维的，具体怎么变呢就是： 假设y是[2，576，1，1]，输入全连接层fc_1之后,pytorch会自动将[2，576，1，1]压成[2X576，1X1]，这就是二维的了。 所以应该改成如下：

在这里插入图片描述

在pytorch中，全连接层是通过torch.nn.linear()这个函数实现的，输入的参数只有feature的channels(就是途中的in_features)，而size则根据前面的量来自适应的，所以很多人会自动的认为不需要注意输入的shape，自适应即可。这时全连接层就会帮你把你超过二维的shape给自动调整成二维的,这时就会报错啦。所以要自己调整fc的输入shape，只要batchSize和Channel数即可。

在这里插入图片描述