RuntimeError: mat1 and mat2 shapes cannot be multiplied (axb and cxd)

这是在全连接层两边维度不匹配所导致的报错。

处理方法是将全连接层中的c更改为报错信息中的b,使b=c,这样两个矩阵就可以做乘法了,从而得到a个(一个batch)中每个图片的d各种类预测分数结果,解决。

你可能感兴趣的:(bug调试,pytorch,python)