pytorch numpy list类型之间的相互转换

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)

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • pytorch笔记
    中文版官方教程：http://pytorch123.com/ 1.一篇总结很好的文章 pytorch学习：http...
    阳光果粒C阅读 1,220评论 1赞 9
  • PyTorch学习笔记(二)：Tensor操作
    什么是Tensor Tensor，又名张量，最早接触这个词是来自于TensorFlow，这个概念也被广泛的应用于不...
    Daisy丶阅读 54,086评论 1赞 20
  • NumPy回顾
    换工作之后，PYTHON就放下啦，基本没接触了，也都忘记的差不多了，现在重新捡起来 1.1 NumPy Array...
    wqh8384阅读 584评论 0赞 1
  • 安史之乱
    唐玄宗李隆基创业的时候，励精图治，君明臣贤；守业的时候，荒淫无度，君昏臣奸。前后判若两人，仿佛做了变性手术似的。 ...
    梅香书宛阅读 473评论 0赞 0
  • 478篇｜下半年的成长
    今天是建党节，也是下半年的第一天，总结了上半年，就应该规划下下半年。 早起：继续坚持早起，持续增加能量，只有自律才...
    Rechel218阅读 249评论 0赞 0