高版本pytorch出现IndexError: invalid index of a 0-dim tensor.问题解决办法

最近使用pytorch1.4版本训练网络，在运用loss.data（）时，目的时为了让loss函数的值相加，但出现问题IndexError: invalid index of a 0-dim tensor.问题解决办法。
经查找，发现问题的原因
loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.
该问题的解决办法：
第一种：
如果不想更改代码，则需要安装低版本的pytorch。
但不推荐这种做法。因为低版本的在现在很多开源新算法中已经被抛弃了。
第二种：
将代码中的

loss+=loss.data[0]

修改为：

train_loss+=loss.item()

改完之后就会愉快的进行网络的训练了。