torch.Tensor.item

Tensor.item() → number

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

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

Пример:

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