参考原链接:(39条消息) 关于python中.item()的用法_千律的博客-优快云博客_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数

本文介绍了Python中PyTorch库的Tensor对象的.item()方法。该方法用于将只包含一个元素的张量转换为Python标量,以获取更高的精度。在进行梯度下降或需要高精度计算的场景下,.item()非常有用。错误示例展示了当尝试从非单元素张量调用.item()时,会抛出ValueError。
1490

被折叠的 条评论
为什么被折叠?



