torch.Tensor.item

Tensor.item() → number

Возвращает значение этого тензора как стандартного числа Python. Это работает только для тензоров с одним элементом. В других случаях см. tolist().

Эта операция не дифференцируема.

Пример:

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