torch.unflatten

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

Расширяет измерение входного тензора по нескольким измерениям.

См. также

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

Параметры

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

Возвращает

Просмотр input с указанным распакованным измерением.

Примеры::

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