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实现linux下使用xcopy的方法
Jun 28 Python
对python3 urllib包与http包的使用详解
May 10 Python
django rest framework 数据的查找、过滤、排序的示例
Jun 25 Python
Python使用ConfigParser模块操作配置文件的方法
Jun 29 Python
mac PyCharm添加Python解释器及添加package路径的方法
Oct 29 Python
Python 中的 global 标识对变量作用域的影响
Aug 12 Python
python生成并处理uuid的实现方式
Mar 03 Python
Python3 利用face_recognition实现人脸识别的方法
Mar 13 Python
python实现对变位词的判断方法
Apr 05 Python
Django权限设置及验证方式
May 13 Python
Python调用.net动态库实现过程解析
Jun 05 Python
Python基础之变量的相关知识总结
Jun 23 Python
Python Pygame实战在打砖块游戏的实现
python超详细实现完整学生成绩管理系统
Mar 17 #Python
Python Pygame实战之塔防游戏的实现
pytorch中的 .view()函数的用法介绍
Mar 17 #Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
Python与C++中梯度方向直方图的实现
You might like
PHP详细彻底学习Smarty
2008/03/27 PHP
php中文字母数字验证码实现代码
2008/04/25 PHP
php数组函数序列之array_slice() - 在数组中根据条件取出一段值,并返回
2011/11/07 PHP
php如何实现不借助IDE快速定位行数或者方法定义的文件和位置
2017/01/17 PHP
jquery简单体验
2007/01/10 Javascript
javascript OFFICE控件测试代码
2009/12/08 Javascript
关于jQuery中的end()使用方法
2011/07/10 Javascript
利用函数的惰性载入提高javascript代码执行效率
2014/05/05 Javascript
由ReactJS的Hello world说开来
2015/07/02 Javascript
node.js利用redis数据库缓存数据的方法
2017/03/01 Javascript
JS获取当前地理位置的方法
2017/10/25 Javascript
基于JavaScript实现幸运抽奖页面
2020/07/05 Javascript
React-router4路由监听的实现
2018/08/07 Javascript
jQuery实现的鼠标拖动浮层功能示例【拖动div等任何标签】
2018/12/29 jQuery
Javascript之高级数组API的使用实例
2019/03/08 Javascript
vue 子组件和父组件传值的示例
2020/09/11 Javascript
vue 动态创建组件的两种方法
2020/12/31 Vue.js
在Python中使用Mako模版库的简单教程
2015/04/08 Python
SVM基本概念及Python实现代码
2017/12/27 Python
Django 浅谈根据配置生成SQL语句的问题
2018/05/29 Python
Python Web编程之WSGI协议简介
2018/07/18 Python
python进行文件对比的方法
2018/12/24 Python
关于python的缩进规则的知识点详解
2020/06/22 Python
python两种获取剪贴板内容的方法
2020/11/06 Python
HTML5 body设置全屏背景图片的示例代码
2020/12/08 HTML / CSS
常见的软件开发流程有哪些
2015/11/14 面试题
什么是方法的重载
2013/06/24 面试题
学前教育专业毕业生自荐信
2013/10/03 职场文书
电气工程和自动化自荐信范文
2013/12/25 职场文书
四年级科学教学反思
2014/02/10 职场文书
学校感恩教育活动总结
2014/07/07 职场文书
推普周国旗下讲话稿
2014/09/21 职场文书
刑事和解协议书范本
2014/11/19 职场文书
小学生心理健康活动总结
2015/05/08 职场文书
小学同学聚会感言
2015/07/30 职场文书
校园安全教育心得体会
2016/01/15 职场文书