【python】list,tensor,numpy相互转化

# 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)