PyTorch中的torch.cat简单介绍


Posted in Python onMarch 17, 2022

1.toych简单介绍

torch包含了多维疑是的数据结构及基于其上的多种数学操作。

torch包含了多维张量的数据结构以及基于其上的多种数学运算。此外,它也提供了多种实用工具,其中一些可以更有效地对张量和任意类型进行序列化的工具。

它具有CUDA的对应实现,可以在NVIDIA GPU上进行张量运算(计算能力>=3.0)

2. 张量Tensors

torch.is_tensor(obj):如果obj是一个pytorch张量,则返回True

torch.is_storage(obj):如果obj是一个pytorch storage对象,则返回True

torch.numel(input):返回input张量中的元素个数。

3.torch.cat

a = torch.ones([1,2])
 
b = torch.ones([1,2])
 
z = torch.cat([a,b],1)
 
a
Out[47]: tensor([[1., 1., 1., 1.]])
 
a
Out[48]: tensor([[1., 1.]])

如果第二个参数是1,torch.cat就是将a,b 按列放在一起,大小为torch.Size([1,4])。如果第二个参数是0,则按行

行放在一起,大小为 torch.Size([2, 2]) 。

字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat是concatenate的意思,即拼接,联系在一起。

例子理解:

import torch
A = torch.ones(2,3)
A
#tensor([[1., 1., 1.],
#        [1., 1., 1.]])
B=2*torch.ones(4,3)
B
#tensor([[2., 2., 2.],
#        [2., 2., 2.],
#        [2., 2., 2.],
#        [2., 2., 2.]])
C = torch.cat((A,B),0) #按维数0(添加到行)拼接
C
#tensor([[1., 1., 1.],
#        [1., 1., 1.],
#        [2., 2., 2.],
#        [2., 2., 2.],
#        [2., 2., 2.],
#        [2., 2., 2.]])
D = 2*torch.ones(2,4)
M = torch.cat((A,D),1)  # 按维数1(列)拼接
M
#tensor([[1., 1., 1., 2., 2., 2., 2.],
#        [1., 1., 1., 2., 2., 2., 2.]])
M.size()
#torch.Size([2, 7])

使用torch.cat((A,B),dim)时,除拼接维数dim数值可不同外其余维数数值需相同,方能对齐

到此这篇关于PyTorch中的torch.cat的文章就介绍到这了,更多相关torch.cat内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python中多线程thread与threading的实现方法
Aug 18 Python
简单理解Python中基于生成器的状态机
Apr 13 Python
Python中生成器和迭代器的区别详解
Feb 10 Python
python3使用smtplib实现发送邮件功能
May 22 Python
对python的文件内注释 help注释方法
May 23 Python
python验证码识别教程之利用投影法、连通域法分割图片
Jun 04 Python
利用Django提供的ModelForm增删改数据的方法
Jan 06 Python
Python 3.8 新功能来一波(大部分人都不知道)
Mar 11 Python
PyCharm+PyQt5+QtDesigner配置详解
Aug 12 Python
python批量提取图片信息并保存的实现
Feb 05 Python
详解OpenCV曝光融合
Apr 29 Python
详解OpenCV获取高动态范围(HDR)成像
Apr 29 Python
Python Pygame实战在打砖块游戏的实现
python超详细实现完整学生成绩管理系统
Mar 17 #Python
Python Pygame实战之塔防游戏的实现
pytorch中的 .view()函数的用法介绍
Mar 17 #Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
Python与C++中梯度方向直方图的实现
You might like
如何冲泡挂耳包咖啡?技巧是什么
2021/03/04 冲泡冲煮
php split汉字
2009/06/05 PHP
PHP MYSQL实现登陆和模糊查询两大功能
2016/02/05 PHP
将PHP程序中返回的JSON格式数据用gzip压缩输出的方法
2016/03/03 PHP
Javascript 类、命名空间、代码组织代码
2011/07/31 Javascript
Json序列化和反序列化方法解析
2013/12/19 Javascript
JS图像无缝滚动脚本非常好用
2014/02/10 Javascript
javascript实现在某个元素上阻止鼠标右键事件的方法和实例
2014/08/12 Javascript
JavaScript实现垂直向上无缝滚动特效代码
2016/11/23 Javascript
原生JS实现跑马灯效果
2017/02/20 Javascript
JavaScript与JQUERY获取元素的宽、高和位置
2017/02/26 Javascript
vue深入解析之render function code详解
2017/07/18 Javascript
vuejs使用递归组件实现树形目录的方法
2017/09/30 Javascript
nodejs 使用nodejs-websocket模块实现点对点实时通讯
2018/11/28 NodeJs
node错误处理与日志记录的实现
2018/12/24 Javascript
js实现指定时间倒计时效果
2019/08/26 Javascript
vue表单数据交互提交演示教程
2019/11/13 Javascript
详解vue-flickity的fullScreen功能实现
2020/04/07 Javascript
vue中watch和computed的区别与使用方法
2020/08/23 Javascript
在Vue中使用mockjs代码实例
2020/11/25 Vue.js
[39:00]Optic vs VP 2018国际邀请赛淘汰赛BO3 第三场 8.24
2018/08/25 DOTA
python启动办公软件进程(word、excel、ppt、以及wps的et、wps、wpp)
2009/04/09 Python
Python实现微信公众平台自定义菜单实例
2015/03/20 Python
python正则表达式match和search用法实例
2015/03/26 Python
python 读写中文json的实例详解
2017/10/29 Python
python 剪切移动文件的实现代码
2018/08/02 Python
python机器学习之KNN分类算法
2018/08/29 Python
python操作日志的封装方法(两种方法)
2019/05/23 Python
美国玩具公司:U.S.Toy
2018/05/19 全球购物
如何写好升职自荐信
2014/01/06 职场文书
干部个人考察材料
2014/12/24 职场文书
保洁员岗位职责
2015/02/04 职场文书
办公室文员岗位职责
2015/02/04 职场文书
罚款通知怎么写
2015/04/22 职场文书
贫困证明怎么写
2015/06/16 职场文书
四群教育工作总结
2015/08/10 职场文书