python中.item的用法

参考原链接：(39条消息) 关于python中.item()的用法_千律的博客-CSDN博客_item()

x = torch.randn(2,2)
print(x) 
print(x[1][1])
print(x[1][1].item())

 

从这里可以看出.item的输出精度更高，所以在梯度下降或者需要高精度的场合可以用。

而且，看下方

print(x.item())

会有提示：ValueError: only one element tensors can be converted to Python scalars

只有一个元素张量可以转换为Python标量

官方小册子是这么写的：使用这个函数可以从一个只包含一个值的张量中得到一个Python数