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 相关文章推荐
使用C语言来扩展Python程序和Zope服务器的教程
Apr 14 Python
python查看FTP是否能连接成功的方法
Jul 30 Python
Python打包文件夹的方法小结(zip,tar,tar.gz等)
Sep 18 Python
Python3爬虫爬取百姓网列表并保存为json功能示例【基于request、lxml和json模块】
Dec 05 Python
Python实现的统计文章单词次数功能示例
Jul 08 Python
Python 用三行代码提取PDF表格数据
Oct 13 Python
python自动生成model文件过程详解
Nov 02 Python
python psutil监控进程实例
Dec 17 Python
如何定义TensorFlow输入节点
Jan 23 Python
Tensorflow中tf.ConfigProto()的用法详解
Feb 06 Python
如何基于Python Matplotlib实现网格动画
Jul 20 Python
Pandas实现批量拆分与合并Excel的示例代码
May 30 Python
Python Pygame实战在打砖块游戏的实现
python超详细实现完整学生成绩管理系统
Mar 17 #Python
Python Pygame实战之塔防游戏的实现
pytorch中的 .view()函数的用法介绍
Mar 17 #Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
Python与C++中梯度方向直方图的实现
You might like
VML绘图板②脚本--VMLgraph.js、XMLtool.js
2006/10/09 PHP
来自PHP.NET的入门教程
2006/10/09 PHP
Http 1.1 Etag 与 Last-Modified提高php效率
2008/01/10 PHP
php去除HTML标签实例
2013/11/06 PHP
PHP中$this和$that指针使用实例
2015/01/06 PHP
YII Framework框架教程之使用YIIC快速创建YII应用详解
2016/03/15 PHP
PHPUnit测试私有属性和方法功能示例
2018/06/12 PHP
Jquery AutoComplete自动完成 的使用方法实例
2010/03/19 Javascript
用jquery模仿的a的title属性(兼容ie6/7)
2013/01/21 Javascript
IE下双击checkbox反应延迟问题的解决方法
2014/03/27 Javascript
实现easyui的datagrid导出为excel的示例代码
2016/11/10 Javascript
NodeJs使用Mysql模块实现事务处理实例
2017/05/31 NodeJs
详解基于node的前端项目编译时内存溢出问题
2017/08/01 Javascript
详解create-react-app 2.0版本如何启用装饰器语法
2018/10/23 Javascript
通过图带你深入了解vue的响应式原理
2019/06/21 Javascript
ES6学习笔记之let与const用法实例分析
2020/01/22 Javascript
ES6扩展运算符和rest运算符用法实例分析
2020/05/23 Javascript
Vue 组件注册全解析
2020/12/17 Vue.js
Python实现TCP/IP协议下的端口转发及重定向示例
2016/06/14 Python
浅谈django开发者模式中的autoreload是如何实现的
2017/08/18 Python
在java中如何定义一个抽象属性示例详解
2017/08/18 Python
python3.6环境安装+pip环境配置教程图文详解
2019/06/20 Python
Python实现微信机器人的方法
2019/09/06 Python
python代码能做成软件吗
2020/07/24 Python
PyCharm 2020.2下配置Anaconda环境的方法步骤
2020/09/23 Python
Html5之title吸顶功能
2018/06/04 HTML / CSS
世界首屈一指的钓鱼用品商店:TackleDirect
2016/07/26 全球购物
John Hardy官方网站:手工设计首饰的奢侈品牌
2017/07/05 全球购物
GOOD AMERICAN官网:为曲线性感而设计
2017/12/28 全球购物
澳大利亚领先的在线礼品网站:Gifts Australia
2020/08/15 全球购物
志愿者宣传口号
2014/06/17 职场文书
公司年夜饭通知
2015/04/25 职场文书
我的兄弟姐妹观后感
2015/06/15 职场文书
大学军训通讯稿(2016最新版)
2015/12/21 职场文书
小数乘法教学反思
2016/02/22 职场文书
SQL Server使用PIVOT与unPIVOT实现行列转换
2022/05/25 SQL Server