Python PyTorch Tensor.tolist用法及代碼示例

本文簡要介紹python語言中 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