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爬取NUS-WIDE数据库图片
Oct 05 Python
Python标准库shutil用法实例详解
Aug 13 Python
利用python提取wav文件的mfcc方法
Jan 09 Python
在Python中字典根据多项规则排序的方法
Jan 21 Python
使用python3构建文件传输的方法
Feb 13 Python
python实现windows倒计时锁屏功能
Jul 30 Python
Python实用库 PrettyTable 学习笔记
Aug 06 Python
Django ORM 自定义 char 类型字段解析
Aug 09 Python
关于Python Tkinter Button控件command传参问题的解决方式
Mar 04 Python
tensorflow使用freeze_graph.py将ckpt转为pb文件的方法
Apr 22 Python
基于python实现计算两组数据P值
Jul 10 Python
Python一些基本的图像操作和处理总结
Jun 23 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
Thinkphp框架+Layui实现图片/文件上传功能分析
2020/02/07 PHP
离开页面时检测表单元素是否被修改,提示保存的js代码
2010/08/25 Javascript
IE6中使用position导致页面变形的解决方案(js代码)
2011/01/09 Javascript
javascript实现点击后变换按钮显示文字的方法
2015/05/13 Javascript
jquery实现可关闭的倒计时广告特效代码
2015/09/02 Javascript
jQuery实现移动端滑块拖动选择数字效果
2015/12/24 Javascript
全面解析bootstrap格子布局
2016/05/22 Javascript
jquery.form.js框架实现文件上传功能案例解析(springmvc)
2016/05/26 Javascript
简单实现nodejs上传功能
2017/01/14 NodeJs
详解离线安装npm包的几种方法
2018/11/25 Javascript
nuxt中使用路由守卫的方法步骤
2019/01/27 Javascript
详解JavaScript 为什么要有 Symbol 类型?
2019/04/03 Javascript
微信小程序实现侧边栏分类
2019/10/21 Javascript
[01:10]DOTA2 Supermajor:英雄,由我们见证
2018/05/14 DOTA
[53:50]CHAOS vs Mineski 2019国际邀请赛小组赛 BO2 第一场 8.16
2019/08/18 DOTA
[48:46]完美世界DOTA2联赛PWL S2 SZ vs FTD.C 第二场 11.19
2020/11/19 DOTA
Python单例模式实例分析
2015/01/14 Python
使用Python监控文件内容变化代码实例
2018/06/04 Python
Pycharm配置远程调试的方法步骤
2018/12/17 Python
django admin后台添加导出excel功能示例代码
2019/05/15 Python
centos 安装Python3 及对应的pip教程详解
2019/06/28 Python
Django实现auth模块下的登录注册与注销功能
2019/10/10 Python
Flask框架 CSRF 保护实现方法详解
2019/10/30 Python
Python3.7实现验证码登录方式代码实例
2020/02/14 Python
基于python实现简单C/S模式代码实例
2020/09/14 Python
Pytorch 图像变换函数集合小结
2021/02/01 Python
HTML5使用Audio标签实现歌词同步的效果
2016/03/17 HTML / CSS
【HTML5】Canvas绘制简单图片教程
2016/05/13 HTML / CSS
HTML5本地存储之IndexedDB
2017/06/16 HTML / CSS
《珍珠泉》教学反思
2014/02/20 职场文书
优秀大学生求职自荐信范文
2014/04/19 职场文书
2014年药店店长工作总结
2014/11/17 职场文书
毕业典礼致辞
2015/07/29 职场文书
导游词之海南-南湾猴岛
2019/10/12 职场文书
redis客户端实现高可用读写分离的方式详解
2021/07/04 Redis
从原生JavaScript到React深入理解
2022/07/23 Javascript