pytorch实现Tensor变量之间的转换


Posted in Python onFebruary 17, 2020

系统默认是torch.FloatTensor类型

data = torch.Tensor(2,3)是一个2*3的张量,类型为FloatTensor

data.cuda()就转换为GPU的张量类型,torch.cuda.FloatTensor类型

(1) CPU或GPU之间的张量转换

在Tensor后加long(), int(), double(),float(),byte()等函数就能将Tensor进行类型转换type()函数,

data为Tensor数据类型,data.type()为给出data的类型,

如果使用data.type(torch.FloatTensor)则强制转换为torch.FloatTensor类型张量

(2) CPU张量转化成GPU张量

data.cuda()

(3) GPU张量转化成CPU张量

data.cpu()

(4) Variable变量转换成普通的Tensor

Variable是一个Wrapper,装在里面的data是tensor,如果Var是Variable变量,使用Var.data获得Tensor变量

(5) Tensor与numpy array之间的转换

Tensor->numpy 使用data.numpy(),data为Tensor变量

Numpy->Tensor 使用torch.from_numpy(data),data为numpy变量

(6) 分别获取张量和数组的尺寸,注意size的使用

torch 张量

获取张量的尺寸 a.size()

numpy 数组

获取数组的尺寸 b.shape

获取数组中元素的个数: b.size (这里和张量中的属性的size的含义不同)

(7) 升维和降维的问题

unsqueeze(N)升维到第N维

squeeze(N)降维第N维

需要做如下操作:

x = x.unsqueeze(0) 假如x=(3,1080,1920) 操作后 x = (1,3,1080,1920)

降维也是做同样的操作

以上这篇pytorch实现Tensor变量之间的转换就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python 时间处理datetime实例
Sep 06 Python
python关键字and和or用法实例
May 28 Python
PyChar学习教程之自定义文件与代码模板详解
Jul 17 Python
python爬虫_实现校园网自动重连脚本的教程
Apr 22 Python
Django框架多表查询实例分析
Jul 04 Python
python实现屏保程序(适用于背单词)
Jul 30 Python
python防止随意修改类属性的实现方法
Aug 21 Python
Pytorch之contiguous的用法
Dec 31 Python
用Python做一个久坐提醒小助手的示例代码
Feb 10 Python
使用Django和Postgres进行全文搜索的实例代码
Feb 13 Python
keras实现基于孪生网络的图片相似度计算方式
Jun 11 Python
详解pandas apply 并行处理的几种方法
Feb 24 Python
Macbook安装Python最新版本、GUI开发环境、图像处理、视频处理环境详解
Feb 17 #Python
PyCharm无法识别PyQt5的2种解决方法,ModuleNotFoundError: No module named 'pyqt5'
Feb 17 #Python
python识别验证码图片实例详解
Feb 17 #Python
Python pyautogui模块实现鼠标键盘自动化方法详解
Feb 17 #Python
Matplotlib使用字符串代替变量绘制散点图的方法
Feb 17 #Python
关于tf.TFRecordReader()函数的用法解析
Feb 17 #Python
将数据集制作成VOC数据集格式的实例
Feb 17 #Python
You might like
坏狼的PHP学习教程之第2天
2008/06/15 PHP
php 上传功能实例代码
2010/04/13 PHP
PHP 文件系统详解
2012/09/13 PHP
实例简介PHP的一些高级面向对象编程的特性
2015/11/27 PHP
thinkphp框架实现删除和批量删除
2016/06/29 PHP
PHP Socket网络操作类定义与用法示例
2017/08/30 PHP
PHP curl批处理及多请求并发实现方法分析
2018/08/15 PHP
PHP从尾到头打印链表实例讲解
2018/09/27 PHP
php微信公众号开发之翻页查询
2018/10/20 PHP
JavaScript 调试器简介
2009/02/21 Javascript
jquery 操作单选框,复选框,下拉列表实现代码
2009/10/27 Javascript
推荐40个简单的 jQuery 导航插件和教程(下篇)
2012/09/14 Javascript
jQuery中has()方法用法实例
2015/01/06 Javascript
js+html5实现可在手机上玩的拼图游戏
2015/07/17 Javascript
jquery按回车键实现表单提交的简单实例
2016/05/25 Javascript
jQuery easyUI datagrid 增加求和统计行的实现代码
2016/06/01 Javascript
js基础之DOM中元素对象的属性方法详解
2016/10/28 Javascript
React中ES5与ES6写法的区别总结
2017/04/21 Javascript
微信小程序-getUserInfo回调的实例详解
2017/10/27 Javascript
JS中DOM元素的attribute与property属性示例详解
2018/09/04 Javascript
jQuery实现的鼠标拖动画矩形框示例【可兼容IE8】
2019/05/17 jQuery
Node Express用法详解【安装、使用、路由、中间件、模板引擎等】
2020/05/13 Javascript
[01:06:07]2014 DOTA2国际邀请赛中国区预选赛5.21 DT VS CIS
2014/05/22 DOTA
解决Python中由于logging模块误用导致的内存泄露
2015/04/23 Python
Python爬虫之模拟知乎登录的方法教程
2017/05/25 Python
Python网络编程 Python套接字编程
2017/09/13 Python
Python实现类似比特币的加密货币区块链的创建与交易实例
2018/03/20 Python
python读取文本中的坐标方法
2018/10/14 Python
python实现控制COM口的示例
2019/07/03 Python
python使用celery实现异步任务执行的例子
2019/08/28 Python
Python脚本打包成可执行文件过程解析
2020/10/20 Python
一款利用纯css3实现的360度翻转按钮的实例教程
2014/11/05 HTML / CSS
如何将一个描述日期或日期/时间的字符串转换为一个Date对象
2015/10/13 面试题
历史专业毕业生的自我鉴定
2013/11/15 职场文书
签约仪式主持词
2014/03/19 职场文书
股份转让协议书
2014/04/12 职场文书