python3　list, np.array, torch.tensor相互转换

ndarray = np.array(list)  # list 转 numpy数组
list = ndarray.tolist()  # numpy 转 list
tensor=torch.Tensor(list)  # list 转 torch.Tensor
list = tensor.numpy().tolist()  # torch.Tensor 转 list  先转numpy，后转list
ndarray = tensor.cpu().numpy()  # torch.Tensor 转 numpy  *gpu上的tensor不能直接转为numpy
tensor = torch.from_numpy(ndarray)  # numpy 转 torch.Tensor
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
文章知识点与官方知识档案匹配，可进一步学习相关知识
Python入门技能树首页概览416686 人正在系统学习中
注：本文转载自blog.csdn.net的黑白德芙Sani的文章"https://blog.csdn.net/qq_29617883/article/details/99771411"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！