torch.Tensor.item

Tensor.item() → number

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

此操作不可微分。

示例

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