欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python

程序员文章站 2024-03-17 11:20:22
...
return tot_acc.data[0], pos_acc.data[0], neg_acc.data[0] #更改前
return tot_acc.item(), pos_acc.item(), neg_acc.item() #更改后

更改一下代码格式就好

相关标签: python 机器学习