Pytorch: list, numpy. Tensor 格式轉化 （附 only one element tensors can be converted to Python scalars 解決）

Pytorch ：list, numpy.array, torch.Tensor 格式相互轉化

同時解決 ValueError:only one element tensors can be converted to Python scalars 問題

 

torch.Tensor 轉 numpy

ndarray = tensor.numpy()

若是是在 gpu，命令以下

ndarray = tensor.cpu().numpy()  # 這是由於 gpu上的 tensor 不能直接轉爲 numpy

 

numpy 轉 torch.Tensor

tensor = torch.from_numpy(ndarray) 

 

list 轉 torch.Tensor

tensor=torch.Tensor(list)

注意：有時，上面操做會出現報錯：ValueError:only one element tensors can be converted to Python scalars

緣由是：要轉換的list裏面的元素包含多維的tensor。

在 gpu 上的解決方法是：

val= torch.tensor([item.cpu().detach().numpy() for item in val]).cuda()

#  這是由於 gpu上的 tensor 不能直接轉爲 numpy； 須要先在 cpu 上完成操做，再回到 gpu 上

# 若是是在 cpu 上，上面的 .cpu() 和 .cuda() 能夠省略

 

torch.Tensor 轉 list

list = tensor.numpy().tolist()   # 先轉 numpy，後轉 list

 

list 轉 numpy

ndarray = np.array(list)

 

numpy 轉 list

list = ndarray.tolist()