loss = loss.unsqueeze(0) 升维
losses.update(loss.item(), loss.size(0))
IndexError: dimension specified as 0 but tensor has no dimensions
a = torch.tensor([1])
b = torch.tensor(1)
a 和 b 之间是有很大的区别的
IndexError: dimension specified as 0 but tensor has no dimensions
PyTorch中张量维度操作详解
最新推荐文章于 2025-03-10 21:15:56 发布
本文详细解析了PyTorch中张量升维与更新操作的常见错误及解决办法,通过对比不同维度的张量,如一维张量和标量,解释了在loss计算与更新过程中的维度匹配问题。
2548

被折叠的 条评论
为什么被折叠?



