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

最新推荐文章于 2023-08-08 10:36:16 发布

一壶浊酒..

最新推荐文章于 2023-08-08 10:36:16 发布

阅读量4.9k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： # pytorch 文章标签： python 深度学习人工智能

本文链接：https://blog.csdn.net/qq_40107571/article/details/126576960

pytorch 专栏收录该内容

57 篇文章 ¥39.90 ¥99.00

订阅专栏

在运行PyTorch代码时遇到`IndexError: invalid index of a 0-dim tensor`的问题。解决方案是避免对0维张量使用索引，如`loss.data[0]`，应改用`loss.item()`。这是因为从PyTorch 0.4版本开始，`loss.data`直接返回张量值，0维张量没有索引。同时，介绍了`.data`和`.detach()`的区别，推荐在访问张量数据时使用`.detach()`以避免影响自动求导过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当运行下面代码时出现下面的错误

if iteration%10 == 0:
        print("===> Epoch[{}]({}/{}): Loss: {:.10f}".format(epoch, iteration, len(training_data_loader), loss.data[0]))

修改建议：

去掉loss.data[0]后面的[0]。

if iteration%10 == 0:
        print("===> Epoch[{}]({}/{}): Loss: {:.10f}".format(epoch, iteration, len(training_data_loader), loss.data))

原因：loss是0维张量

创建一个Tensor默认是不可导的，即 .requeires_grad 属性等于false。

在pytorch0.4版本之后，Variable和Tensor进行的合并。loss.data直接输出tensor值，不输出tensor的梯度信息。比如：

loss

了解本专栏