torch.Tensor.item

Tensor.item() → number

將此張量的值作為標準的 Python 數字返回。這僅適用於單元素張量。對於其他情況，請參閱 tolist()。

此操作不可微分。

示例

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