torch.flip

torch.flip(input, dims) → Tensor

Обращает порядок элементов N-мерного тензора вдоль указанных осей в dims.

Примечание

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

Параметры:

• input (Tensor) – входной тензор.
• dims (список или tuple) – оси, вдоль которых нужно выполнить обращение

Пример:

>>> 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]]])