开发者社区

文档建议反馈控制台

最新优惠活动

文章/答案/技术大牛

发布

社区首页 >专栏 >机器学习笔记之list, numpy.array, torch.Tensor 格式相互转化

机器学习笔记之list, numpy.array, torch.Tensor 格式相互转化

作者头像

Jetpropelledsnake21

发布于 2020-03-17 14:07:30

1.7K0

发布于 2020-03-17 14:07:30

举报

文章被收录于专栏：JetpropelledSnake

0x00 list 转 numpy

ndarray = np.array(list)

0x01 numpy 转 list

list = ndarray.tolist()

0x02 list 转 torch.Tensor

tensor=torch.Tensor(list)

0x03 torch.Tensor 转 list

先转numpy，后转list

list = tensor.numpy().tolist()

0x04 torch.Tensor 转 numpy

ndarray = tensor.numpy()

*gpu上的tensor不能直接转为numpy

ndarray = tensor.cpu().numpy()

0x05 numpy 转 torch.Tensor

tensor = torch.from_numpy(ndarray)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020-03-13 ，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

评论

登录后参与评论

0 条评论

热度

最新

LV.

目录

0x00 list 转 numpy
0x01 numpy 转 list
0x02 list 转 torch.Tensor
0x03 torch.Tensor 转 list
0x04 torch.Tensor 转 numpy
0x05 numpy 转 torch.Tensor