快捷方式

torch.Tensor.tolist

Tensor.tolist() list 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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源