问题1 : mat1 dim1 must match mat2 dim=0
通过问题的描述发现错误应该在线性层上(Linear)上发生,输入矩阵与权重矩阵的尺寸不匹配。
可能的解决办法
1.检查数据的维度是否正确输入网络模型,确定输入数据各个维度之间的大小以及是否与模型期望的大小匹配。
2.确保一个矩阵的行数等于另一个矩阵的列数,以便它们相乘,也就是说检查哪个矩阵的第二维度不匹配。
3.可以使用调试器或print语句来确定张量的尺寸,以找到输入数据中的错误。
4.使用torch,size()函数来检查输入Tensor的形状,例如输出x的尺寸
print(x.size())