Pytorch:invalid index of a 0-dim tensor.Use tensor.item() to convert a 0-dim tensor to a Python的解决方法

Pytorch: invalid index of a 0-dim tensor.Use tensor.item() to convert a 0-dim tensor to a Python number 在学习“白话强化学习PyTorch”时遇到的错误(原代码如下): 运行时系统报错: 解决方法: 将"loss.data[0]" 更改为"loss.data",问题就解决了! 这样程序就能正常执
相关文章
相关标签/搜索