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