評價此頁

torch.Tensor.item#

Tensor.item() number#

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

此操作不可微分。

示例

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