Python中 list, numpy.array, torch.Tensor 格式相互轉化

1.1 list 轉 numpy

ndarray = np.array(list)

 

1.2 numpy 轉 list

list = ndarray.tolist()

 

2.1 list 轉 torch.Tensor

tensor=torch.Tensor(list)

 

2.2 torch.Tensor 轉 list

先轉numpy,後轉list

list = tensor.numpy().tolist()

 

3.1 torch.Tensor 轉 numpy

ndarray = tensor.numpy()

*gpu上的tensor不能直接轉爲numpy

ndarray = tensor.cpu().numpy()

 

3.2 numpy 轉 torch.Tensor

tensor = torch.from_numpy(ndarray) 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章