torch.flatten

torch.flatten(input, start_dim=0, end_dim=- 1) → Tensor

Расплющивает input, преобразуя его в одномерный тензор. Если start_dim или end_dim указаны, сплющиваются только размерности, начиная с start_dim и заканчивая end_dim. Порядок элементов в input не изменяется.

В отличие от NumPy’s flatten, который всегда копирует данные входного объекта, эта функция может вернуть исходный объект, представление или копию. Если размерности не сплющиваются, то возвращается исходный объект input. В противном случае, если входной объект можно представить в виде сплющенной формы, возвращается это представление. И только если входной объект нельзя представить в виде сплющенной формы, данные входного объекта копируются. Подробности о том, когда будет возвращено представление, см. в torch.Tensor.view().

Примечание

Расплющивание нульмерного тензора вернёт одномерное представление.

Параметры:

• input (Тензор) – входной тензор.
• start_dim (int) – первая размерность для сплющивания
• end_dim (int) – последняя размерность для сплющивания

Пример:

>>> t = torch.tensor([[[1, 2],
...                    [3, 4]],
...                   [[5, 6],
...                    [7, 8]]])
>>> torch.flatten(t)
tensor([1, 2, 3, 4, 5, 6, 7, 8])
>>> torch.flatten(t, start_dim=1)
tensor([[1, 2, 3, 4],
        [5, 6, 7, 8]])