torch.unbind

torch.unbind(input, dim=0) → seq

Удаляет размерность тензора.

Возвращает кортеж всех срезов вдоль заданной размерности, уже без неё.

Параметры:

• input (Тензор) – тензор для удаления размерности
• dim (int) – размерность для удаления

Пример:

>>> torch.unbind(torch.tensor([[1, 2, 3],
>>>                            [4, 5, 6],
>>>                            [7, 8, 9]]))
(tensor([1, 2, 3]), tensor([4, 5, 6]), tensor([7, 8, 9]))