RuntimeError: The size of tensor a (64) must match the size of tensor b (44) at non-singleton dimension 0
时间: 2023-11-06 11:42:05 浏览: 123
IOS Object-C 中Runtime详解及实例代码
这个错误通常是因为两个张量尺寸不匹配导致的。你需要检查你的代码,找到这个错误出现的地方,然后确保两个张量的尺寸相同。你可以使用 PyTorch 的 size() 或 numel() 函数来检查张量的尺寸。另外,你也可以试着使用 PyTorch 的 resize() 或 view() 函数来调整张量的尺寸使其匹配。
阅读全文