pytorch tensor合并与分割

1. cat

torch.cat(tensors, dim=0, *, out=None) → Tensor
在指定维度上,连接给定tensor序列或empty,除连接的dimension外,所有得的ensor必须有相同的shape
参数:
tensors-具有相同类型的tensor序列,非empty tensor必须具有相同的shape,连接的dimension除外
dim-指定的连接的维度
输出:
连接后的tensor

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

上图分别是在列和行两个维度连接后的结果

2. stack

创建新维度来连接张量序列

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

参数:
tensors-张量序列,具有相同的size
dim-插入的新维度,必须介于0和连接的tensor的维度之间
输出:
连接后的tensor
在这里插入图片描述
注意:cat和stack的区别
stack连接的tensor必须具有相同的size,否则报错,cat是除连接的维度外,其他维度shape必须相同
如下示例:
在这里插入图片描述

3. split

把一个tensor切分成块,每个块是原tensor的一部分

torch.split(tensor, split_size_or_sections, dim=0)

参数:
tensor-用来切分的tensor
split_size_or_sections (int) or (list(int)) -单个块的size后者是每个块size的list
dim (int) – 以tensor的哪个维度进行切分
输出:
Tuple[Tensor, …]

示例:
在这里插入图片描述

4. chunk

强制将一个tensor切分成指定数量的块,每个块是原tensor的一部分

torch.chunk(input, chunks, dim=0) → List of Tensors

参数:
input (Tensor) – 输入切分的tensor
chunks (int) – 切分块的数量
dim (int) – 以tensor的哪个维度进行切分
输出:
切分后的list

示例:
在这里插入图片描述
注意:split与chunk的区别
区别主要是第二个参数,split第二个参数切分块的size,而chunk是切分块的数量

相关推荐

  1. 文件分割合并

    2024-02-20 09:36:04       38 阅读
  2. vue实现文件分割合并

    2024-02-20 09:36:04       16 阅读
  3. Linux分割合并文件

    2024-02-20 09:36:04       37 阅读
  4. PDF分割合并

    2024-02-20 09:36:04       11 阅读
  5. linux tar分卷压缩windows合并解压

    2024-02-20 09:36:04       24 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-02-20 09:36:04       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-02-20 09:36:04       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-02-20 09:36:04       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-02-20 09:36:04       20 阅读

热门阅读

  1. Go json Marshal & UnMarshal 的一点小 trick

    2024-02-20 09:36:04       33 阅读
  2. 【LeetCode-494】目标和(回溯&动归)

    2024-02-20 09:36:04       28 阅读
  3. 第13章 网络 Page749~755 asio核心类 ip::tcp::acceptor

    2024-02-20 09:36:04       27 阅读
  4. vue小记——this

    2024-02-20 09:36:04       33 阅读
  5. mac下mysql 常用命令

    2024-02-20 09:36:04       28 阅读
  6. PaddleSeg分割框架解读[01] 核心设计解析

    2024-02-20 09:36:04       25 阅读
  7. vue中使用jsx语法

    2024-02-20 09:36:04       26 阅读