pytorch中数据转换
1. CPU tensor转GPU tensor：
1 | cpu_data.cuda() |
2. GPU tensor 转CPU tensor：
1 | gpu_data.cpu() |
3. numpy转为CPU tensor：
1 | torch.from_numpy(np_data) |
4.CPU tensor转为numpy数据：
1 | cpu_data.numpy() |
注：
1.note：GPU tensor不能直接转为numpy数组，必须先转到CPU tensor。
2.torch.from_numpy(array)是做数组的浅拷贝，torch.Tensor(array)是做数组的深拷贝（浅拷贝映射指针；深拷贝完全复制）转载请注明来源，欢迎对文章中的引用来源进行考证，欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论，也可以邮件至 luckiestman0@qq.com
文章标题:pytorch中数据转换
文章字数:89
本文作者:Dashan
发布时间:2020-07-18, 16:30:26
最后更新:2020-07-18, 16:37:04
原始链接:http://yoursite.com/2020/07/18/pytorch中数据转换/版权声明: "署名-非商用-相同方式共享 4.0" 转载请保留原文链接及作者。
