当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.Tensor.tolist。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。