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

Posted 2021-11-03 siyuan1998

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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)

以上是关于Python中 list, numpy.array, torch.Tensor 格式相互转化的主要内容，如果未能解决你的问题，请参考以下文章