item()函数是得到一个元素张量里面的元素值。
下面我们来举一个例子:
>>> x = torch.randn(1)
>>> print(x)
tensor([0.6328])
>>> print(x.item())
0.632796049118042
item()函数是得到一个元素张量里面的元素值。
下面我们来举一个例子:
>>> x = torch.randn(1)
>>> print(x)
tensor([0.6328])
>>> print(x.item())
0.632796049118042