torch.chunk

torch.chunk(input: Tensor, chunks: int, dim: int = 0) → Tuple[Tensor, ...]

Пытается разделить тензор на указанное количество частей. Каждая часть является представлением входного тензора.

Примечание

Эта функция может вернуть меньше частей, чем было указано!

См. также

torch.tensor_split() — функция, которая всегда возвращает ровно указанное количество частей

Если размер тензора вдоль заданного измерения dim делится на chunks без остатка, все возвращённые части будут одинакового размера. Если размер тензора вдоль заданного измерения dim не делится на chunks без остатка, все возвращённые части, кроме последней, будут одинакового размера. Если такое разделение невозможно, эта функция может вернуть меньше частей, чем было указано.

Параметры:

• input (Tensor) – тензор для разделения
• chunks (int) – количество частей для возврата
• dim (int) – измерение, вдоль которого нужно разделить тензор

Пример

>>> torch.arange(11).chunk(6)
(tensor([0, 1]),
 tensor([2, 3]),
 tensor([4, 5]),
 tensor([6, 7]),
 tensor([8, 9]),
 tensor([10]))
>>> torch.arange(12).chunk(6)
(tensor([0, 1]),
 tensor([2, 3]),
 tensor([4, 5]),
 tensor([6, 7]),
 tensor([8, 9]),
 tensor([10, 11]))
>>> torch.arange(13).chunk(6)
(tensor([0, 1, 2]),
 tensor([3, 4, 5]),
 tensor([6, 7, 8]),
 tensor([ 9, 10, 11]),
 tensor([12]))