【pytorch】.item()的用法

用法描述

Use torch.Tensor.item() to get a Python number from a tensor containing a single value.

.item()方法返回张量元素的值。

用法示例

>>> import torch
>>> x = torch.tensor([[1]])
>>> x
tensor([[1]])
>>> x.item()
1
>>> x = torch.tensor(2.5)
>>> x
tensor(2.5000)
>>> x.item()
2.5

注意事项

张量中只有一个元素才能调用该方法，多个元素会报错：

>>> import torch
>>> x = torch.tensor([1, 2]) 
>>> x
tensor([1, 2])
>>> x.item()
Traceback (most recent call last):
  File "", line 1, in 
ValueError: only one element tensors can be converted to Python scalars

引用参考

https://pytorch.org/docs/stable/tensors.html