size mismatch, m1: [450 x 1], m2: [30 x 1] at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensorMath.cpp:136
时间: 2023-09-21 15:11:09 浏览: 74
pytorch逐元素比较tensor大小实例
这个错误通常是因为在矩阵乘法时,两个矩阵的维度不匹配导致的。请检查你的代码,确保矩阵的维度是匹配的。
比如说,在 PyTorch 中,两个矩阵的乘法需要满足以下条件:
- 第一个矩阵的列数(即第二个维度)和第二个矩阵的行数(即第一个维度)相等。
- 输出的矩阵的维度是第一个矩阵的行数(即第一个维度)和第二个矩阵的列数(即第二个维度)。
如果你的两个矩阵的维度不满足上述条件,就会出现 size mismatch 的错误。你可以打印出两个矩阵的形状并检查它们是否满足上述条件。
阅读全文