torch.Tensor.tolist

Tensor.tolist() → list 或 number

將張量 (tensor) 回傳為（巢狀）列表 (list)。對於純量 (scalar)，則會回傳標準 Python 數值，如同使用 item() 一樣。如有必要，張量會先自動移至 CPU。

此操作不可微分。

範例

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