IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or `tensor.item＜T＞()`解决办法

遇到问题如下：

File "/root/yi/cyclegan/PyTorch -CycleGAN mas ter/utils.py", line 39, in log

self.losses[loss name] = losses [loss name] .data[0]

IndexError: invalid index of a 0-dim tensor. Usetensor .item(): in Python or tensor. item-T>() in C++o convert a 0-dim tensor to a number

解决方法：

self.losses[loss name] = losses [loss name] .data[0]

替换为：

self.losses[loss name] = losses [loss name] .item()

注意：所有 .data[0] 均要替换成为 .item()