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

问题描述

使用Pytorch框架调试论文的代码时，报 IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convert a 0-dim tensor to a number 错误，如下图所示：
报错代码如下所示：

解决方法

将上述的报错代码

print('%.4f' % errG.data[0])

修改为：

print('%.4f' % errG.item())

修改后再次运行，不再报错。

参考资料

https://blog.csdn.net/qq_42255269/article/details/108287251