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正则分组的应用
Nov 10 Python
安装dbus-python的简要教程
May 05 Python
Perl中著名的Schwartzian转换问题解决实现
Jun 02 Python
Django objects.all()、objects.get()与objects.filter()之间的区别介绍
Jun 12 Python
Python实现希尔排序算法的原理与用法实例分析
Nov 23 Python
python操作excel的包(openpyxl、xlsxwriter)
Jun 11 Python
python实现定时提取实时日志程序
Jun 22 Python
python3 http提交json参数并获取返回值的方法
Dec 19 Python
Python 实现王者荣耀中的敏感词过滤示例
Jan 21 Python
Python 堆叠柱状图绘制方法
Jul 29 Python
Python对接 xray 和微信实现自动告警
Sep 17 Python
解决pycharm编辑区显示yaml文件层级结构遇中文乱码问题
Apr 27 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
投票管理程序
2006/10/09 PHP
PHP安全配置详细说明
2011/09/26 PHP
解析php时间戳与日期的转换
2013/06/06 PHP
php使用类继承解决代码重复的问题
2015/02/11 PHP
java微信开发之上传下载多媒体文件
2016/06/24 PHP
简单谈谈PHP面向对象之标识对象
2017/06/27 PHP
php-msf源码详解
2017/12/25 PHP
关于PHP虚拟主机概念及如何选择稳定的PHP虚拟主机
2018/11/20 PHP
如何在Laravel5.8中正确地应用Repository设计模式
2019/11/26 PHP
Javascript YUI 读码日记之 YAHOO.util.Dom - Part.3
2008/03/22 Javascript
一个可以随意添加多个序列的tag函数
2009/07/21 Javascript
javascript for循环设法提高性能
2010/02/24 Javascript
jQuery的12招常用技巧分享
2011/08/08 Javascript
javascript 判断整数方法分享
2014/12/16 Javascript
jQuery 利用$.ajax 时获取原生XMLHttpRequest 对象的方法
2016/08/25 Javascript
浅谈javascript中的事件冒泡和事件捕获
2016/12/28 Javascript
jQuery实现淡入淡出的模态框
2017/02/09 Javascript
解决vue-cli项目webpack打包后iconfont文件路径的问题
2018/09/01 Javascript
vue实现的上拉加载更多数据/分页功能示例
2019/05/25 Javascript
JS实现扫码枪扫描二维码功能
2020/01/03 Javascript
VUE中setTimeout和setInterval自动销毁案例
2020/09/07 Javascript
Python中多线程及程序锁浅析
2015/01/21 Python
Python 由字符串函数名得到对应的函数(实例讲解)
2017/08/10 Python
python正则实现计算器功能
2017/12/14 Python
python画柱状图--不同颜色并显示数值的方法
2018/12/13 Python
python 实现矩阵上下/左右翻转,转置的示例
2019/01/23 Python
详解pandas中MultiIndex和对象实际索引不一致问题
2019/07/23 Python
python获取天气接口给指定微信好友发天气预报
2020/12/28 Python
Python Selenium操作Cookie的实例方法
2021/02/28 Python
广告学专业求职信
2014/06/19 职场文书
校园元旦活动总结
2014/07/09 职场文书
竞选班干部演讲稿500字
2014/08/20 职场文书
2014年建筑工程工作总结
2014/12/03 职场文书
2016年教师党员创先争优承诺书
2016/03/24 职场文书
python办公自动化之excel的操作
2021/05/23 Python
MySQL中IF()、IFNULL()、NULLIF()、ISNULL()函数的使用详解
2021/06/26 MySQL