invalid index of a 0-dim tensor, Use tensor.item() to convert a 0-dim tensor to a number 报错

转载：https://blog.csdn.net/weixin_41797117/article/details/80237179
将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告，它在提醒用户下个版本这将成为一个错误
修改方法如下：

#原语句：
train_loss+=loss.data[0]
#修改后：
train_loss+=loss.item()
#bingo