RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 271 but got size 0 for tensor number 1 in the list.
时间: 2023-11-06 20:40:43 浏览: 72
pytorch dataloader 取batch_size时候出现bug的解决方式
这个错误通常出现在PyTorch中,它表示在进行某些操作时,两个张量(tensors)的大小不匹配,除了维度1以外。具体来说,它表明在一个张量中有一个维度的大小为0,而另一个张量相应的维度的大小为271,这导致了两者不匹配。建议检查张量的大小和形状,并确保它们在进行操作时是兼容的。还可以使用PyTorch提供的调试工具(如torch.set_grad_enabled)来找出哪个计算步骤出了问题。
阅读全文