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

用pytorch时会出现这种情况：

test_loss += criterion(output, target).data[0]

这条语句会报错：

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

anaconda会提示用item()替换

改为：

test_loss += criterion(output, target).item()

Python0.4.0以上的版本中loss是一个零维的标量。对标量进行索引是没有意义的，所以使用loss.item()可以从标量中获取Python数字。