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

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

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是0维张量，因此在输出loss.data的时候维度为0，可以用loss.data.size()测试。可以利用loss.item()将张量转换为python number。