torch.flip

torch.flip(input, dims) → Tensor

Инвертирует порядок элементов n-мерного тензора вдоль указанных осей в dims.

Примечание

torch.flip создает копию данных input. Это отличается от NumPy np.flip, который возвращает представление за константное время. Поскольку копирование данных тензора требует больше вычислений, чем получение представления, torch.flip ожидается, что будет медленнее, чем np.flip.

Параметры

• input (Tensor) – входной тензор.
• dims (список или кортеж) – ось для инвертирования

Пример:

>>> x = torch.arange(8).view(2, 2, 2)
>>> x
tensor([[[ 0,  1],
         [ 2,  3]],

        [[ 4,  5],
         [ 6,  7]]])
>>> torch.flip(x, [0, 1])
tensor([[[ 6,  7],
         [ 4,  5]],

        [[ 2,  3],
         [ 0,  1]]])