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的Django框架生成PDF文件的教程
Jul 22 Python
Python中datetime模块参考手册
Jan 13 Python
Python cookbook(数据结构与算法)同时对数据做转换和换算处理操作示例
Mar 23 Python
ubuntu17.4下为python和python3装上pip的方法
Jun 12 Python
python爬取Ajax动态加载网页过程解析
Sep 05 Python
Python 进程操作之进程间通过队列共享数据,队列Queue简单示例
Oct 11 Python
PyCharm永久激活方式(推荐)
Sep 22 Python
IntelliJ 中配置 Anaconda的过程图解
Jun 01 Python
详解anaconda安装步骤
Nov 23 Python
用python对excel查重
Dec 07 Python
python不同版本的_new_不同点总结
Dec 09 Python
Python进程间的通信之语法学习
Apr 11 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中current、next与reset函数用法实例
2014/11/17 PHP
PHP利用正则表达式将相对路径转成绝对路径的方法示例
2017/02/28 PHP
Laravel学习教程之View模块详解
2017/09/18 PHP
PHP 文件上传限制问题
2019/09/01 PHP
Prototype使用指南之hash.js
2007/01/10 Javascript
js event事件的传递与冒泡处理
2009/12/06 Javascript
jQuery获取节点和子节点文本的方法
2014/07/22 Javascript
每天一篇javascript学习小结(基础知识)
2015/11/10 Javascript
JavaScript设置名字输入不合法的实现方法
2017/05/23 Javascript
js实现图片粘贴上传到服务器并展示的实例
2017/11/08 Javascript
JS+Canvas绘制动态时钟效果
2017/11/10 Javascript
React中使用外部样式的3种方式(小结)
2019/05/28 Javascript
layui实现数据分页功能(ajax异步)
2019/07/27 Javascript
js动态生成表格(节点操作)
2021/01/12 Javascript
Python3.0与2.X版本的区别实例分析
2014/08/25 Python
VSCode Python开发环境配置的详细步骤
2019/02/22 Python
Python创建或生成列表的操作方法
2019/06/19 Python
Python+OpenCV+pyQt5录制双目摄像头视频的实例
2019/06/28 Python
python与C、C++混编的四种方式(小结)
2019/07/15 Python
Python爬取腾讯视频评论的思路详解
2019/12/19 Python
快速解决pymongo操作mongodb的时区问题
2020/12/05 Python
Python3 用matplotlib绘制sigmoid函数的案例
2020/12/11 Python
探究 canvas 绘图中撤销(undo)功能的实现方式详解
2018/05/17 HTML / CSS
阿迪达斯新加坡官方网站:adidas新加坡
2019/12/06 全球购物
加拿大领先家居家具网上购物:Aosom.ca
2020/05/27 全球购物
J2SDK1.5与J2SDK5.0有什么区别
2012/09/19 面试题
《燕子》教学反思
2014/02/18 职场文书
小学生手册家长评语
2014/04/16 职场文书
校园环保建议书
2014/05/14 职场文书
说好普通话圆梦你我他演讲稿
2014/09/21 职场文书
2015年人事科工作总结
2015/04/28 职场文书
张丽莉事迹观后感
2015/06/16 职场文书
交通事故责任认定书
2015/08/06 职场文书
2016年学校党支部公开承诺书
2016/03/25 职场文书
spring cloud eureka 服务启动失败的原因分析及解决方法
2022/03/17 Java/Android
详解Python中__new__方法的作用
2022/03/31 Python