torch.item()和torch.data的区别
当tensor中只有一个值的时候 .item()返回一个数值,.data返回一个tensor
使用例:
test = torch.Tensor(42)
a = test.item()
b = test.data
print(f'a:{a}, b:{b}')
#输出为:a:42, b:tensor(42)
另
torch 低版本的单值tensor不会自动转换为数值类型,
为保证兼容性,求和后尽量使用.item