废话不多,方法简单:RuntimeError: mat1 and mat2 shapes cannot be multiplied (1x16 and 20x64)
这是非常常见的问题,主要有以下两种情况:1.卷积层和全连接层之间的参数数量不对应,两个张量(矩阵)相乘,一定是第一个矩阵的列数=第二个矩阵的行数。2.全连接神经网络:当前网络结构输出的神经元数和设置的神经元不匹配,因此需要重新设置fc层的神经元数这里我们主要说第二种情况!原代码:classFCNNModel(torch.nn.Module):def__init__(self,input_num,h