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",問題就解決了! 這樣程序就能正常執
相關文章
相關標籤/搜索