解决【IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item＜T＞()`】

项目场景：

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

---

问题描述

报错：IndexError: invalid index of a 0-dim tensor. Use tensor.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