RuntimeError: size mismatch, m1: [8 x 6], m2: [5 x 32] at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensorMath.cpp:752
时间: 2023-08-12 20:05:53 浏览: 169
jdk-8u251-windows-x64.zip
这个错误是由于PyTorch中矩阵的尺寸不匹配引起的。在你的代码中,你正在尝试将一个大小为[8 x 6]的矩阵与一个大小为[5 x 32]的矩阵相乘,但是两个矩阵的尺寸不一致,因此会引发该错误。
要解决这个问题,你需要确保矩阵相乘的维度匹配。在这种情况下,你需要调整矩阵的尺寸,使其能够相乘。你可以使用PyTorch中的函数来改变矩阵的形状,如`view()`或`reshape()`。
例如,如果你想将大小为[8 x 6]的矩阵m1与大小为[5 x 32]的矩阵m2相乘,你可以使用`view()`函数将m1重塑为[48 x 6]的矩阵,然后与m2相乘。
```python
m1 = m1.view(48, 6)
result = torch.mm(m1, m2)
```
请注意,在进行矩阵乘法之前,确保你已经调整了矩阵的形状以匹配相乘的维度。
阅读全文