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 之间是有很大的区别的。
本文分享自 作者个人站点/博客 前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!