torch.Tensor.item

Tensor.item() → 数字

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

这个操作不可微分。

示例：

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