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中signal包的使用
Nov 13 Python
python编程实现归并排序
Apr 14 Python
基于Django filter中用contains和icontains的区别(详解)
Dec 12 Python
pyqt5 实现 下拉菜单 + 打开文件的示例代码
Jun 20 Python
Python pandas DataFrame操作的实现代码
Jun 21 Python
python求绝对值的三种方法小结
Dec 04 Python
爬虫代理池Python3WebSpider源代码测试过程解析
Dec 20 Python
pyinstaller还原python代码过程图解
Jan 08 Python
Python如何获取Win7,Win10系统缩放大小
Jan 10 Python
Python是什么 Python的用处
May 26 Python
Python基于time模块表示时间常用方法
Jun 18 Python
Python调用高德API实现批量地址转经纬度并写入表格的功能
Jan 12 Python
Python Pygame实战在打砖块游戏的实现
python超详细实现完整学生成绩管理系统
Mar 17 #Python
Python Pygame实战之塔防游戏的实现
pytorch中的 .view()函数的用法介绍
Mar 17 #Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
Python与C++中梯度方向直方图的实现
You might like
用PHP实现验证码功能
2006/10/09 PHP
基于PHP静态类的原罪详解
2013/05/06 PHP
深入PHP FTP类的详解
2013/06/13 PHP
php引用传值实例详解学习
2013/11/06 PHP
浅析Yii2 gridview实现批量删除教程
2016/04/22 PHP
万能的php分页类
2017/07/06 PHP
PHP7扩展开发之基于函数方式使用lib库的方法详解
2018/01/15 PHP
YII框架实现自定义第三方扩展操作示例
2019/04/26 PHP
javascript实现连续赋值
2015/08/10 Javascript
Jquery常用的方法汇总
2015/09/01 Javascript
JavaScript自定义浏览器滚动条兼容IE、 火狐和chrome
2017/01/05 Javascript
jQuery实现滚动条滚动到子元素位置(方便定位)
2017/01/08 Javascript
Bootstrap 中data-[*] 属性的整理
2018/03/13 Javascript
webpack4 升级迁移的实现
2018/09/12 Javascript
Vue2.5学习笔记之如何在项目中使用和配置Vue
2018/09/26 Javascript
JavaScript变量基本使用方法实例分析
2019/11/15 Javascript
antd table按表格里的日期去排序操作
2020/11/17 Javascript
如何使用 JavaScript 操作浏览器历史记录 API
2020/11/24 Javascript
原生js实现九宫格拖拽换位
2021/01/26 Javascript
python中的__slots__使用示例
2015/02/26 Python
Hadoop中的Python框架的使用指南
2015/04/22 Python
python对数组进行反转的方法
2015/05/20 Python
Python科学画图代码分享
2017/11/29 Python
Python 新建文件夹与复制文件夹内所有内容的方法
2018/10/27 Python
Python字符串、列表、元组、字典、集合的补充实例详解
2019/12/20 Python
英国川宁茶官方网站:Twinings茶
2019/05/21 全球购物
在C语言中"指针和数组等价"到底是什么意思?
2014/03/24 面试题
Java的类可以定义为Protected或者Private得吗
2015/09/25 面试题
计算机专业毕业生自荐信
2013/12/31 职场文书
小学生期末评语
2014/04/21 职场文书
真诚的求职信
2014/07/04 职场文书
单位租房协议范本
2014/12/03 职场文书
2014年纪委工作总结
2014/12/05 职场文书
2014年后备干部工作总结
2014/12/08 职场文书
关于运动会的宣传稿
2015/07/23 职场文书
Django如何创作一个简单的最小程序
2021/05/12 Python