RuntimeError: The size of tensor a (4) must match the size of tensor b (22) at non-singleton dimension 0
时间: 2023-08-14 11:47:05 浏览: 80
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
5星 · 资源好评率100%
这个错误通常是由于在进行张量运算时,两个张量的形状不兼容导致的。在该错误信息中,张量a的形状是(4,),而张量b的形状是(22,),这意味着它们在第0个维度上的大小不相同。你需要检查你的代码,确保在进行张量运算时两个张量的形状是兼容的。你可以使用PyTorch中的.view()方法来调整张量的形状,以确保它们是兼容的。
阅读全文