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