RuntimeError: The size of tensor a (60) must match the size of tensor b (56) at non-singleton dimension 3
时间: 2023-09-28 21:08:36 浏览: 223
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
5星 · 资源好评率100%
这个错误是由于在某个维度上,两个张量的尺寸不匹配导致的。在你的情况下,张量a具有60个元素,而张量b具有56个元素,且在第3个维度上尺寸不一样。为了解决这个问题,你需要确保这两个张量在所有维度上的尺寸都是相同的。你可以使用适当的函数或方法来调整张量的大小,或者在创建这两个张量时就确保它们具有相同的尺寸。
阅读全文