lossVal += losstensor.data[0] IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

解釋：如果是torch0.3.x版本的代碼在torch0.4.x版本環(huán)境下跑，會(huì)報(bào)出如上錯(cuò)誤，應(yīng)作如下修改

原代碼：lossVal += losstensor.data[0] 修改后：lossVal += losstensor.item()

?

總結(jié)

以上是生活随笔為你收集整理的一个torch版本报错的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。