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 (Tensor) – входной тензор.
• 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]])