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不带重复的全排列代码
Aug 13 Python
Python中函数的用法实例教程
Sep 08 Python
Python判断文件或文件夹是否存在的三种方法
Jul 27 Python
python 识别图片中的文字信息方法
May 10 Python
Python3+django2.0+apache2+ubuntu14部署网站上线的方法
Jul 07 Python
python3通过selenium爬虫获取到dj商品的实例代码
Apr 25 Python
python 将字符串完成特定的向右移动方法
Jun 11 Python
Python Django 实现简单注册功能过程详解
Jul 29 Python
PIL.Image.open和cv2.imread的比较与相互转换的方法
Jun 03 Python
Python实现手绘图效果实例分享
Jul 22 Python
Python txt文件如何转换成字典
Nov 03 Python
pytorch实现线性回归以及多元回归
Apr 11 Python
Python Pygame实战在打砖块游戏的实现
python超详细实现完整学生成绩管理系统
Mar 17 #Python
Python Pygame实战之塔防游戏的实现
pytorch中的 .view()函数的用法介绍
Mar 17 #Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
Python与C++中梯度方向直方图的实现
You might like
Laravel5.7框架安装与使用学习笔记图文详解
2019/04/02 PHP
php传值和传引用的区别点总结
2019/11/19 PHP
document.all还是document.getElementsByName?
2006/07/21 Javascript
js左侧多级菜单动态的解决方案
2010/02/01 Javascript
Jquery倒数计时按钮setTimeout的实例代码
2013/07/04 Javascript
javascript通过className来获取元素的简单示例代码
2014/01/10 Javascript
Nodejs中读取中文文件编码问题、发送邮件和定时任务实例
2015/01/01 NodeJs
简述AngularJS相关的一些编程思想
2015/06/23 Javascript
jQuery表单验证简单示例
2016/10/17 Javascript
详解Vue自定义过滤器的实现
2017/01/10 Javascript
jQuery.cookie.js实现记录最近浏览过的商品功能示例
2017/01/23 Javascript
jQuery监听浏览器窗口大小的变化实例
2017/02/07 Javascript
JavaScript数组和对象的复制
2017/03/21 Javascript
小程序实现左右来回滚动字幕效果
2018/12/28 Javascript
详解JavaScript中的函数、对象
2019/04/01 Javascript
详解javascript对数组和json数组的操作
2019/04/15 Javascript
Vue scoped及deep使用方法解析
2020/08/01 Javascript
JavaScript实现单点登录的示例
2020/09/23 Javascript
python监测当前联网状态并连接的实例
2018/12/18 Python
python爬虫 爬取58同城上所有城市的租房信息详解
2019/07/30 Python
在Sublime Editor中配置Python环境的详细教程
2020/05/03 Python
Django中F函数的使用示例代码详解
2020/07/06 Python
python中的unittest框架实例详解
2021/02/05 Python
德国最大的设计师鞋网上商店:Budapester
2017/12/07 全球购物
印度尼西亚手表和包包商店:Urban Icon
2019/12/12 全球购物
如何在.net Winform里面显示PDF文档
2012/09/11 面试题
AJAX检测用户名是否存在的方法
2021/03/24 Javascript
班级读书活动总结
2014/06/30 职场文书
欢度春节标语
2014/07/01 职场文书
医学专业大学生求职信
2014/07/12 职场文书
教师个人成长总结
2015/02/11 职场文书
2015年南京大屠杀纪念日活动总结
2015/03/24 职场文书
2015年综治维稳工作总结
2015/04/07 职场文书
基层党建工作简报
2015/07/21 职场文书
入门学习Go的基本语法
2021/07/07 Golang
Nebula Graph解决风控业务实践
2022/03/31 MySQL