問題描述

使用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