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之入门(四)运算
May 27 Python
Python中使用logging模块打印log日志详解
Apr 05 Python
python实现决策树
Dec 21 Python
python实现m3u8格式转换为mp4视频格式
Feb 28 Python
Python版名片管理系统
Nov 30 Python
python之cv2与图像的载入、显示和保存实例
Dec 05 Python
关于pytorch中网络loss传播和参数更新的理解
Aug 20 Python
Python中logging日志库实例详解
Feb 19 Python
一文了解python 3 字符串格式化 F-string 用法
Mar 04 Python
Tensorflow tf.tile()的用法实例分析
May 22 Python
Python爬虫基础讲解之请求
May 13 Python
Python 读写 Matlab Mat 格式数据的操作
May 19 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
discuz Passport 通行证 整合笔记
2008/06/30 PHP
php Static关键字实用方法
2010/06/04 PHP
CodeIgniter安全相关设置汇总
2014/07/03 PHP
php类中的$this,static,final,const,self这几个关键字使用方法
2015/12/14 PHP
JavaScript中的History历史对象
2008/01/16 Javascript
JavaScript 继承的实现
2009/07/09 Javascript
jQuery ul标签下拉菜单演示代码
2010/12/11 Javascript
js 走马灯简单实例
2013/11/21 Javascript
改变隐藏的input中value值的方法
2014/03/19 Javascript
Javascript常用字符串判断函数代码分享
2014/12/08 Javascript
基于jquery实现发送文章到手机的代码
2014/12/26 Javascript
浅谈Jquery为元素绑定事件
2015/04/27 Javascript
node.js+express制作网页计算器
2016/01/17 Javascript
JS获取多维数组中相同键的值实现方法示例
2017/01/06 Javascript
引入外部js脚本加载慢与页面白屏问题的解决
2018/12/10 Javascript
vue动态子组件的两种实现方式
2019/09/01 Javascript
js神秘的电报密码 哈弗曼编码实现
2019/09/10 Javascript
JS通用方法触发点击事件代码实例
2020/02/17 Javascript
Vue实现鼠标经过文字显示悬浮框效果的示例代码
2020/10/14 Javascript
[04:13]2018国际邀请赛典藏宝瓶Ⅱ饰品一览
2018/07/21 DOTA
详解Python的Django框架中的模版继承
2015/07/16 Python
Python3中的列表,元组,字典,字符串相关知识小结
2017/11/10 Python
Python进程池Pool应用实例分析
2019/11/27 Python
Pytorch之finetune使用详解
2020/01/18 Python
如何以Winsows Service方式运行JupyterLab
2020/08/30 Python
只要五步 就可以用HTML5/CSS3快速制作便签贴特效(图)
2012/06/04 HTML / CSS
Bench加拿大官方网站:英国城市服装品牌
2017/11/03 全球购物
澳大利亚领先的运动鞋商店:Hype DC
2018/03/31 全球购物
美国浴缸、水槽和水龙头购物网站:Vintage Tub & Bath
2019/11/05 全球购物
Envie de Fraise意大利:法国网上推出的孕妇装品牌
2020/10/18 全球购物
如何将字串String转换成整数int
2015/02/21 面试题
卫校中专生个人自我评价
2013/09/19 职场文书
党员承诺书格式
2014/05/21 职场文书
初中作文评语
2014/12/25 职场文书
大学生入党群众意见书
2015/06/02 职场文书
中学政教处工作总结
2015/08/13 职场文书