​ torch.chunk(tensor, chunks, dim)​

torch.chunk函数用于将张量在指定维度上分割成多个块,chunks参数定义了块的数量。例如,将一个3x3的张量按列分成3块,会得到3个1x3的张量;按行分成2块,会得到2个3x2的张量。当块数为偶数且张量维度为奇数时,最后一块可能会包含多一列的数据。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

 torch.chunk(tensor, chunks, dim)

含义:在给定维度上对所给张量进行分块。

参数解释

  • Tensor         -- 要进行分快的张量
  • chunks        -- 分块个数
  • dim              -- 维度,按照此维度进行分块

如下代码:

import torch
x = torch.randn(3, 3)
x
tensor([[ 0.5149,  1.0009, -0.7242],
        [-0.9385, -0.1157, -1.6772],
        [ 1.0187, -2.3512,  1.0539]])


torch.chunk(x, 3, dim = 0)
(tensor([[ 0.5149,  1.0009, -0.7242]]),
 tensor([[-0.9385, -0.1157, -1.6772]]),
 tensor([[ 1.0187, -2.3512,  1.0539]]))


torch.chunk(x, 3, dim = 1)
(tensor([[ 0.5149],
         [-0.9385],
         [ 1.0187]]),
 tensor([[ 1.0009],
         [-0.1157],
         [-2.3512]]),
 tensor([[-0.7242],
         [-1.6772],
         [ 1.0539]]))


torch.chunk(x, 2, dim = 1)       # 注意当chunks为偶数时,而原始张量为所分维数为奇数时的变化
(tensor([[ 0.5149,  1.0009],
         [-0.9385, -0.1157],
         [ 1.0187, -2.3512]]),
 tensor([[-0.7242],
         [-1.6772],
         [ 1.0539]]))


# 另外也可以这么使用
x.chunk(2, dim = 1)
(tensor([[ 0.5149,  1.0009],
         [-0.9385, -0.1157],
         [ 1.0187, -2.3512]]),
 tensor([[-0.7242],
         [-1.6772],
         [ 1.0539]]))

### PyTorch `split` 和 `chunk` 函数的区别 #### 功能描述 PyTorch 的 `split` 和 `chunk` 都用于分割张量,但两者的工作方式有所不同。 对于 `torch.split(tensor, split_size_or_sections, dim=0)` 函数而言,此方法允许指定要切割成的小片段大小或是各个部分的具体尺寸列表。如果提供的是单个整数,则表示每一部分的长度;如果是元组或列表,则精确指定了各分片的大小[^1]。 ```python import torch tensor_example = torch.arange(8) # 使用 split 方法按每份两个元素来划分张量 result_split = torch.split(tensor_example, 2) print([t.tolist() for t in result_split]) ``` 另一方面,`torch.chunk(tensor, chunks, dim=0)` 则更倾向于平均分配输入张量到指定数量的部分中去。当无法均匀切分时,最后一块可能会小于其他部分。这里只接受一个参数来定义希望得到多少个子集。 ```python # 使用 chunk 方法将张量分为三等分 result_chunk = torch.chunk(tensor_example, 3) print([t.tolist() for t in result_chunk]) ``` 这两种操作都支持通过设置维度参数 (`dim`) 来控制沿哪个轴执行拆分,默认情况下是在第零维上工作。 #### 输出差异展示 上述代码会分别打印出由 `split` 和 `chunk` 返回的结果: - 对于 `split`: 如果传入了合适的总和等于原始张量长度的分割尺寸数组,那么将会获得完全按照给定规格被分开的新张量集合; - 而对于 `chunk`: 不管怎样都会尝试尽可能公平地把原数据分成所请求的数量级,即使这意味着某些部分可能比其他的稍大一些或小一点。 因此,在实际编程过程中可以根据具体需求选择合适的方法来进行张量的操作处理。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值