详解pytorch tensor和ndarray转换相关总结
更新时间：2020年09月03日 10:03:21 作者：再困也得吃
这篇文章主要介绍了详解pytorch tensor和ndarray转换相关总结，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
在使用pytorch的时候，经常会涉及到两种数据格式tensor和ndarray之间的转换，这里总结一下两种格式的转换：
1. tensor cpu 和tensor gpu之间的转化：
tensor cpu 转为tensor gpu：
tensor_gpu = tensor_cpu.cuda()
>>> tensor_cpu = torch.ones((2,2))
tensor([[1., 1.],
[1., 1.]])
>>> tensor_gpu = tensor_cpu.cuda()
tensor([[1., 1.],
[1., 1.]], device='cuda:0')
tensor gpu 转为tensor cpu：
tensor_cpu = tensor_gpu.cuda()
>>> tensor_gpu.cpu()
tensor([[1., 1.],
[1., 1.]])
2. tensor cpu 和 ndarray 之间的转化：
tensor cpu 转为 ndarray：
>>> np_array= tensor_cpu.numpy()
array([[1., 1.],
[1., 1.]], dtype=float32)
ndarray 转为 tensor cpu：
注：ndarray的默认精度为64位，Tensor的默认精度位32位，所以通过Tensor直接转换的话，精度会转换到32位，若通过from_numpy的方式，则会保留原来64位精度
>>> torch.from_numpy(np.ones((2,2)))
tensor([[1., 1.],
[1., 1.]], dtype=torch.float64)
>>> torch.Tensor(np.ones((2,2)))
tensor([[1., 1.],
[1., 1.]])
3. tensor cpu 和 scalar 之间的转化：
如果只是训练了一个简单的分类网络，对单个样本的输出会是一个标量（scalar）
>>>torch.ones((1,1)).item() 1.0
通过一张图说明三者的转化方式：
到此这篇关于详解pytorch tensor和ndarray转换相关总结的文章就介绍到这了,更多相关pytorch tensor和ndarray转换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
在图像处理中，直方图是非常重要，也是非常有用的一个处理要素。这篇文章主要介绍了python数字图像处理实现直方图与均衡化，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-05-05
Python Pygame实战之实现经营类游戏梦想小镇代码版
作为一名模拟经营类游戏的发烧友，各种农场类、医院类、铁路类的游戏玩儿了很多年。今天用代码给大家打造一款梦想小镇游戏，希望大家喜欢啦2022-12-12
好久都没玩过愤怒的小鸟了,今天咱自己做一个玩玩,文中有非常详细的代码示例,对想玩的小伙伴们很有用哦,需要的朋友可以参考下2021-06-06
今天小编就为大家分享一篇python 获得任意路径下的文件及其根目录的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
JSON(JavaScript Object Notation, JS 对象简谱) 是一种轻量级的数据交换格式。本文重点给大家介绍python中的json，感兴趣的朋友跟随小编一起看看吧2018-10-10
今天小编就为大家分享一篇python list格式数据excel导出方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
这篇文章主要介绍了Python中encode()方法的使用简介,是Python入门中的基础知识,需要的朋友可以参考下2015-05-05
想要从文字海洋中捕获珍珠般的古诗文吗?通过Python爬取古诗文网,你将掌握直捣黄龙的技能,简单几步,让经典文学尽收囊中,为你的项目增添灵气,需要的朋友可以参考下2024-03-03
这篇文章主要介绍了Python如何爬取qq音乐歌词到本地,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
python基于OpenCV模块实现视频流数据切割为图像帧数据(流程分析)
这篇文章主要介绍了python基于OpenCV模块实现视频流数据切割为图像帧数据,这里今天主要是实践一下视频流数据的预处理工作，需要的朋友可以参考下2022-05-05
最新评论