torch.chunk

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

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

Примечание

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

См. также

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

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

Параметры:

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

Пример::

>>> 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]))