IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> 【Pytorch】cat() 与 stack() 的区别 -> 正文阅读

[人工智能]【Pytorch】cat() 与 stack() 的区别

cat():

torch.cat(tensors, dim=0, *, out=None) → Tensor
连接给定维度中给定的张量序列seq。所有张量必须具有相同的形状(连接维度除外)或为空。
torch.cat()可以看作torch.split() 和 torch.chunk() 的逆运算。

参数

  • tensors (sequence of Tensors) – 除了要进行cat的维度外,其余维度要和待拼接的tensor保持一致。
  • dim (int, optional) – 连接张量的维度。

例子:

>>> x = torch.randn(2, 3)
>>> x
tensor([[ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497]])
>>> torch.cat((x, x, x), 0)
tensor([[ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497],
        [ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497],
        [ 0.6580, -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497]])
>>> torch.cat((x, x, x), 1)
tensor([[ 0.6580, -1.0969, -0.4614,  0.6580, -1.0969, -0.4614,  0.6580,
         -1.0969, -0.4614],
        [-0.1034, -0.5790,  0.1497, -0.1034, -0.5790,  0.1497, -0.1034,
         -0.5790,  0.1497]])

stack():

torch.stack(tensors, dim=0, *, out=None) → Tensor
沿新维度连接一系列张量。
所有张量都需要具有相同的大小。

如对两个 (1, 2) 维的 tensor 在第 0 个维度上 stack,则会变为 (2, 1, 2) 的 tensor;在第 1 个维度上 stack,则会变为 (1, 2, 2) 的 tensor。

参数

  • tensors (sequence of Tensors) – 要连接的tensor
  • dim (int, optional) – 进行连接的维度。

例子:

>>> x = torch.randn(2, 3)
>>> x
tensor([[ 1.4481,  0.9733, -0.2550],
        [-1.1247,  0.2517,  0.3983]])
>>> torch.stack((x,x),0)
tensor([[[ 1.4481,  0.9733, -0.2550],
         [-1.1247,  0.2517,  0.3983]],

        [[ 1.4481,  0.9733, -0.2550],
         [-1.1247,  0.2517,  0.3983]]])
>>> torch.stack((x,x),1)
tensor([[[ 1.4481,  0.9733, -0.2550],
         [ 1.4481,  0.9733, -0.2550]],

        [[-1.1247,  0.2517,  0.3983],
         [-1.1247,  0.2517,  0.3983]]])

区别

torch.cat的使用方法,两个tensor除了要进行cat的维度外,其余维度必须一致,最终生成的tesnor的维度也没有变化。
从上边两个例子,可以看出对于torch.stack来说,会先将原始数据维度扩展一维,然后再按照维度进行拼接,具体拼接操作同torch.cat类似。

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2022-05-18 17:38:31  更:2022-05-18 17:41:07 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 4:48:57-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码