torch.flip

torch.flip(input, dims) → Tensor

沿給定的維度（dims）反轉 n 維張量的順序。

注意

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