python中list-numpy.array-torch.tensor互相转换

旧城等待, 2021-06-22 15:37 634阅读 0赞

⭐️list 转 numpy

  • ndarray = np.array(list)

⭐️list 转 Tensor

  • tensor = torch.Tensor(list)

⭐️numpy 转 list

  • list = ndarray.tolist()

⭐️numpy 转 Tensor

  • tensor = torch.from_numpy(ndarray)

⭐️Tensor 转 list

  • 需要先转为numpy,再转为list

    • list = tensor.numpy().tolist()

⭐️Tensor 转 numpy

  • ndarray = tensor.numpy()
  • 如果tensor再GPU上需要先转到cpu,再转为numpy

    • ndarray = tensor.cpu().numpy()

发表评论

表情:
评论列表 (有 0 条评论,634人围观)

还没有评论,来说两句吧...

相关阅读

    相关 tree list 互相转换

    本人没有亲测过,有问题的话会重新编辑文档,这篇文章里的代码是积分下载换来的,如果涉及侵权联系我。。。。。看到我这边博客的人可以免积分了。。。。。。 import _