python中.item的用法

作者：盐析白兔 | 2024-04-17 10:43:36

踩

.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数

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/439780