Python PyTorch flip用法及代碼示例

本文簡要介紹python語言中 torch.flip 的用法。

用法:

torch.flip(input, dims) → Tensor

參數：

• input(Tensor) -輸入張量。

• dims(一個列表或者tuple) -要翻轉的軸

沿給定軸反轉 n-D 張量的順序，以暗淡為單位。

注意

torch.flip 複製 input 的數據。這與 NumPy 的 np.flip 不同，後者以恒定時間返回視圖。由於複製張量的數據比查看該數據需要更多的工作，因此 torch.flip 預計會比 np.flip 慢。

例子：

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