项目场景：

描述：pytorch航班预测，输出批次训练结果

问题描述

报错：IndexError: invalid index of a 0-dim tensor. Usetensor.item()in Python ortensor.item＜T＞()

原因分析：

提示：torch版本不同

print('Epoch;{},Loss:{:.5f}'.format(e+1,loss.data[0]))

解决方案：

提示：修改loss.data[0]为loss.item()

print('Epoch;{},Loss:{:.5f}'.format(e+1,loss.item()))

参考链接

感谢：https://blog.csdn.net/qq_42255269/article/details/108287251