Pytorch--Tensor, Numpy--Array,Python--List 相互之间的转换

 

1、 Python--List  \rightarrow Numpy--Array

list = [[1, 2, 4, 8], [0.1, 0.4, 0.9, 0.8]]
list_np = np.array(list)
print('list_np的维度:',list_np.shape)
print('list_np的类型:',type(list_np))

运行结果:

 

2、Python--List \rightarrow  Pytorch--Tensor

test = [[1, 2, 4, 8], [0.1, 0.4, 0.9, 0.8]]
tensor_ = torch.Tensor(test)
print('test的类型:', type(test))
print('tensor_的类型:', type(tensor_))

运行结果:

你可能感兴趣的:(Pytorch)