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), -1, (2, 2, 3, 1, 1)).shape
torch.Size([5, 2, 2, 3, 1, 1, 3])