The size of tensor a (16) must match the size of tensor b (18) at non-singleton dimension 3
时间: 2023-09-25 11:17:15 浏览: 68
Python库 | torch-tensor-type-0.0.1.tar.gz
这个错误提示意味着你在尝试将两个不同大小的张量进行某种操作,但是这两个张量在某个非单例维度上的大小不匹配。
在你的情况下,你可能正在尝试对两个张量进行某种操作,比如相加、相减或者拼接等,但是这两个张量在第 3 维上的大小不同。其中一个张量在第 3 维上的大小为 16,而另一个张量在第 3 维上的大小为 18。这样的操作是不允许的,因为在非单例维度上的大小必须匹配。
要解决这个问题,你需要检查你的代码,找到这些张量的来源,并确保它们在需要匹配大小的维度上具有相同的大小。你可以使用 PyTorch 提供的函数,例如 `torch.Size()` 或 `torch.view()` 来检查和修改张量的大小。
阅读全文