pytorch tensor数据类型转换为python数据

一、item()

pytorch tensor数据类型转换为python数据_第1张图片
input:

x = torch.tensor([1.0])
x.item()

output:

1.0

二、tolist()

pytorch tensor数据类型转换为python数据_第2张图片
input:

a = torch.randn(2, 2)
a.tolist()
a[0,0].tolist()

output:

[[0.012766935862600803, 0.5415473580360413],
 [-0.08909505605697632, 0.7729271650314331]]

0.012766935862600803

你可能感兴趣的:(python,pytorch,人工智能)