torch.Tensor.tolist

Tensor.tolist() → list or number

Возвращает тензор как (вложенный) список. Для скаляров возвращается стандартное число Python, как и в случае с item(). Тензоры автоматически перемещаются на процессор (CPU), если это необходимо.

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

Примеры:

>>> a = torch.randn(2, 2)
>>> a.tolist()
[[0.012766935862600803, 0.5415473580360413],
 [-0.08909505605697632, 0.7729271650314331]]
>>> a[0,0].tolist()
0.012766935862600803