torch.Tensor.item

Tensor.item() → number

将此张量的值作为标准 Python 数字返回。这仅适用于包含一个元素的张量。对于其他情况，请参阅 tolist()。

此操作不可微分。

示例

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