torch.flatten

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

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

В отличие от функции flatten в NumPy, которая всегда копирует данные входного объекта, эта функция может вернуть исходный объект, представление или копию. Если измерения не преобразуются, возвращается исходный объект 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]])