【pytorch】torch.cat()与torch.stack()的区别_tensor.cat_Xavier Jiezou的博客-程序员秘密

技术标签: pytorch  

torch.cat()

torch.cat(tensors, dim=0, *, out=None) → Tensor

描述

在给定维度连接给定的张量序列。所有张量必须要么具有相同的形状(要连接的维度除外),要么为空。

参数

  • tensors (张量序列) – 相同类型张量的任何python序列。提供的非空张量必须具有相同的形状,要拼接的维度除外。
  • dim (int, 可选) – 张量连接的维度;默认是0

示例

>>> import torch
>>> a = torch.rand(2, 3)
>>> b = torch.rand(2, 3)
>>> c = torch.cat((a, b), dim=0)
>>> c.shape
torch.Size([4, 3])
>>> c = torch.cat((a, b), dim=1)
>>> c.shape
torch.Size([2, 6])

torch.stack()

torch.stack(tensors, dim=0, *, out=None) → Tensor

描述

沿着一个的维度连接一个张量序列。所有张量都必须具有相同的大小。

参数

  • tensors (张量序列) – 要连接的张量序列。
  • dim (int) – 插入的维度。必须在0和连接张量的维数之间(包含)。

示例

>>> import torch
>>> a = torch.rand(1, 3)
>>> b = torch.rand(1, 3)
>>> c = torch.stack((a, b), dim=0)
>>> c.shape
torch.Size([2, 1, 3])
>>> d = torch.stack((a, b), dim=1)
>>> d.shape
torch.Size([1, 2, 3])
>>> e = torch.stack((a, b), dim=2)
>>> e.shape
torch.Size([1, 3, 2])

引用参考

https://pytorch.org/docs/stable/generated/torch.cat.html
https://pytorch.org/docs/stable/generated/torch.stack.html

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_42951560/article/details/115433442

智能推荐

ognl.MethodFailedException: Method "setXX" failed for object_-droidcoffee-的博客-程序员秘密

由于数据类型不匹配而导致的异常ognl.MethodFailedException: Method "setId" failed for object [email protected] [java.lang.NoSuchMethodException: setId([Ljava.lang.String;)] at ognl.OgnlRunti

Plop小型脚手架工具使用说明_大丁37821的博客-程序员秘密

小型的脚手架工具, 用于快速创建文件具体使用步骤安装 yarn add plop --dev在项目的根目录下新建入口文件 plopfile.js在plopfile.js文件中定义脚手架任务编写用于生成特定类型文件的模板运行脚手架 yarn plop component(生成器的名称) // Plop入口文件,需要导出一个函数 // 此函数接受一个Plop对象,用于创建生成器 module.exports = plop => { plop.setGenera

Solr or Lucene全文检索实现原理_漫天雪_昆仑巅的博客-程序员秘密

Solr是一个独立的企业级搜索应用服务器,它对外提供类似于Web-service的API接口。用户可以通过http请求,向搜索引擎服务器提交一定格式的XML文件,生成索引;也可以通过Http Get操作提出查找请求,并得到XML/Json格式的返回结果。采用Java5开发,基于Lucene。  Lucene是apache软件基金会4 jakarta项目组的一个子项目,是一个开放源代码的全文

LeetCode 刷题指南以及常见算法题解题思路总结_蔚1的博客-程序员秘密

程序员面试过程中,算法和数据结构是必备知识点之一。无论大公司还是小公司,多多少少都会涉及算法题目。LeetCode 成为快速复习算法题目的有力武器,但是 LeetCode 日益增加的题库,短时间内不可能刷完,如何有效的刷题呢?常见的题目有哪些?通过 LeetCode 上面的题目,我们是否能够总结出一类题目的解决方法,从而达到举一反三的效果呢?本文主要面向算法爱好者或者需要准备面试算法的程序员同学...

a10 linux 烧写,A10/A20 Bootloader加载过程分析_findtea的博客-程序员秘密

上面罗嗦了这么多,其实就是为了将uboot和kernel烧写到TF卡上并能够启动,OK,让我们先从分区开始:A20 芯片上电启动的时候,会读取SD卡最前面的 1M 内容,从而得到 bootloader,所以我们需要把 u-boot 写到SD卡的前1M区间。其中详细的SD卡布局如下:起始大小用途08KB存放分区表等内容824KBSPL loader32512KBu-boot544128KBenvir...

汇编语言+IDA安装问题解决汇总_野九的博客-程序员秘密

利用汇编语言计算机和人类链接更为便捷如下图所示:寄存器:简单讲就是CPU可以存储数据的器件,一个CPU可以有多个寄存器。AX、BX是两个不同的寄存器。16位处理器有14个寄存器:AX、BX、CX、DX、SI、DI、SP、BP、IP、CS、SS、DS、ES、PSW。32位处理器有16个寄存器:EAX、EBX、ECX、EDX、ESI、EDI、EBP、ESP、EIP、ES、CS、SS、DS、F...

随便推点

Struts2项目实战 微云盘(一):项目分析_struts2项目源码百度网盘_壮飞的博客-程序员秘密

一、项目简述 进入github下载本项目(开源) 点击这里下载完整源码压缩包(1币) 这是一个模仿网盘的项目,主要是为了熟悉Struts2框架,该项目不涉及底层DAO操作,只涉及到Struts2的相关知识,底层DAO操作是使用数据单元进行模拟的。本项目涉及到的知识有:网页布局、action的实现与配置、输入校验、树、Struts2的AJAX支持、Filter、文件上传下载。实现的功能有: 1

+1阿姆达尔定律_dengdun6257的博客-程序员秘密

阿姆达尔定律可以用来计算处理器平行运算之后效率提升的能力。阿姆达尔定律因Gene Amdal 在1967年提出这个定律而得名。绝大多数使用并行或并发系统的开发者有一种并发或并行可能会带来提速的感觉,甚至不知道阿姆达尔定律。不管怎样,了解阿姆达尔定律还是有用的。我会首先以算术的方式介绍阿姆达尔定律定律,然后再用图表演示一下。阿姆达尔定律定义总结如下:T = B + (...

wpa_supplicant.conf 配置文件解析(一)_二十岁了还没有去过星巴克的博客-程序员秘密

WPA,是Wi-Fi Protected Access,Wi-Fi安全访问的简称。wpa_supplicant是开源项目源码,被谷歌修改后加入android移动平台,它主要是用来支持WEP,WPA/WPA2和WAPI无线协议和加密认证的。WPA_Supplicant功能:通过socket(不管是wpa_supplicant与上层还是wpa_supplicant与驱动都采用socket通讯)与驱动...

python入门_Bank_Lei的博客-程序员秘密

python入门几个概念1.表达式表达式就是一个类似于数学公式的东西比如:10 + 5 8 - 4表达式一般仅仅用了计算一些结果,不会对程序产生实质性的影响如果在交互模式中输入一个表达式,解释器会自动将表达式的结果输出2.语句在程序中语句一般需要完成某种功能,比如打印信息、获取信息、为变量赋值。。。比如:print()input()a = 10语句的执行一般会...

华为服务器怎么进系统还原,服务器怎么还原操作系统_洛斯里克甜面包的博客-程序员秘密

服务器怎么还原操作系统 内容精选换一换执行chmod -R 777 / 导致CentOS云服务器根目录权限设置成777,系统中的大部分服务以及命令无法使用。此时可通过系统自带的getfacl命令来拷贝和还原系统权限,本节操作介绍误操作导致根目录设置成777权限的补救措施。对文件或目录设置777权限代表它将对所有用户都是可读、可写和可执行的,根目录设置777权限后存在安全风险。切换操作系统是为您的弹...

SpringBoot实现动态定时任务_springboot动态添加定时任务_我滴太阳233的博客-程序员秘密

项目情况:         在当前项目中需要一个定时任务来清除过期的校验码,如果使用数据库存储过程的话不方便维护。因此采用SpringBoot自带的方式来设置定时任务。技术说明:         SpringBoot自带的方式有两种可以实现:        一种是使用@Scheduled注解的方式,只需要在启动类或者它所在的类上添加@EnableScheduling注解允许执行定时任务,并且设置S...

推荐文章

热门文章

相关标签