Python报错:IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in

Python报错如下:

复制代码
E:\anaconda\envs\pytorch_gpu\python.exe D:/project/graphsage/graphsage/model.py 
D:\project\graphsage\graphsage\encoders.py:31: UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform_.
  init.xavier_uniform(self.weight)
D:/project/graphsage/graphsage/model.py:28: UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform_.
  init.xavier_uniform(self.weight)
Traceback (most recent call last):
  File "D:/project/graphsage/graphsage/model.py", line 181, in <module>
    run_cora()
  File "D:/project/graphsage/graphsage/model.py", line 102, in run_cora
    print(batch, loss.data[0])
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

Process finished with exit code 1

这是一个Python程序的运行错误信息,提示了以下问题:

NO.1 UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform_.:这是一个警告提示,提醒你在使用nn.init.xavier_uniform时应改为使用nn.init.xavier_uniform_。

NO.2 Traceback (most recent call last):...:这部分是一个错误追踪,显示了程序运行过程中出现错误的调用堆栈。

NO.3 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:这是一个IndexError错误,说明在代码中尝试使用了一个0维张量的索引,而0维张量不支持这种操作。它提供了解决方案,可以使用tensor.item()方法将0维张量转换为数值。

为了解决这些问题,你可以按照以下步骤进行处理:

1 将代码中使用的nn.init.xavier_uniform替换为nn.init.xavier_uniform_,以消除警告。

2 在出现IndexError的地方,使用tensor.item()方法来获取0维张量的数值。

例如,对于init.xavier_uniform(self.weight),你可以将其修改为init.xavier_uniform_(self.weight),并在第102行的loss.data[0]改为loss.item()。

注意:loss.data[0]的操作在较新版本的PyTorch中可能已被弃用,建议使用loss.item()来获取损失值。

本人本地修改如下:

javascript 复制代码
import torch.nn.init as init

# ... 在代码其他部分 ...

# 在encoders.py中的第31行和第28行
init.xavier_uniform_(self.weight)

# 在model.py中的第102行
print(batch, loss.item())

这样修改后,再次运行程序,能解决这些问题。

相关推荐
凉茶钱10 分钟前
【吃透C++】万字解析类和对象
开发语言·c++
白杨尚青33 分钟前
C++入门篇(十):string(上)——认识string:构造与三大遍历(一条龙讲透operator[]、迭代器、auto、范围for)
java·开发语言·c++·笔记·stl
可乐鸡翅yeah_38 分钟前
hls.js 切换多个视频源,新手开发常见踩坑
开发语言·前端·javascript·ios·ffmpeg·音视频·safari
Tanshu_API君41 分钟前
API 聚合平台选型指南:从参数拆解到生产落地的十个评估维度
开发语言·api
Chen—LSN42 分钟前
C语言——⽂件操作(1)
c语言·开发语言·c++·经验分享·笔记·算法·链表
狗凯之家源码网1 小时前
PHP 开源 IM 即时通讯系统效果实测与功能展示
开发语言·开源·php
虎虎(_ _)。゜zzZ1 小时前
关于字典中的解包问题
python
辛苦才能1 小时前
C++ map/set 深度解析:从关联式容器的本质到 operator[] 的三重身份
开发语言·c++
全栈弄潮儿²⁰²⁴1 小时前
Python实战第2期: 变量与数据类型
python·变量·数据类型·python实战
Chen—LSN1 小时前
C语言——文件操作(2)
c语言·开发语言·c++·经验分享·笔记·算法·c#