RuntimeError: The size of tensor a (3) must match the size of tensor b (28) at non-singleton dimension 3
时间: 2023-12-13 14:33:04 浏览: 164
IOS Object-C 中Runtime详解及实例代码
这个报错通常是由于两个张量在非单例维度上的大小不匹配导致的。解决方法是检查两个张量的形状并确保它们在需要匹配的维度上具有相同的大小。可以使用`torch.Size()`函数检查张量的形状,使用`torch.view()`函数调整张量的形状。例如,如果张量a的形状为(3, 4, 5, 6),张量b的形状为(2, 3, 4, 28),则可以使用以下代码将张量b的形状调整为(2, 1, 1, 28)以匹配张量a的形状:
```python
import torch
a = torch.randn(3, 4, 5, 6)
b = torch.randn(2, 3, 4, 28)
if a.shape[0] != b.shape[1]:
b = b.view(b.shape[0], 1, 1, b.shape[3])
c = torch.matmul(a, b)
```
阅读全文