torch.Tensor.item

Tensor.item() → number

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

此操作不可微分。

示例：

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