torch.Tensor.item

Tensor.item() → number

將此 Tensor 的值作為標準 Python 數值傳回。這僅適用於具有一個元素的 Tensor。對於其他情況，請參閱 tolist()。

此操作不可微分。

範例

>>> x = torch.tensor([1.0])
>>> x.item()
1.0