torch.unflatten

torch.unflatten(input, dim, sizes) → Tensor

Раскрывает одно измерение входного тензора в несколько измерений.

См. также

torch.flatten() — обратная функция. Она объединяет несколько измерений в одно.

Параметры:

• input (Tensor) – входной тензор.
• dim (int) – измерение, которое требуется раскрыть, заданное как индекс в input.shape.
• sizes (Tuple[int]) – новая форма раскрытого измерения. Один из элементов может быть -1; в этом случае соответствующее выходное измерение определяется автоматически. В противном случае произведение sizes должно равняться input.shape[dim].

Возвращает:

Представление входного тензора с раскрытым указанным измерением.

Примеры::

>>> torch.unflatten(torch.randn(3, 4, 1), 1, (2, 2)).shape
torch.Size([3, 2, 2, 1])
>>> torch.unflatten(torch.randn(3, 4, 1), 1, (-1, 2)).shape
torch.Size([3, 2, 2, 1])
>>> torch.unflatten(torch.randn(5, 12, 3), -2, (2, 2, 3, 1, 1)).shape
torch.Size([5, 2, 2, 3, 1, 1, 3])