Python PyTorch flipud用法及代碼示例

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

用法:

torch.flipud(input) → Tensor

參數：

input(Tensor) -必須至少是一維的。

向上/向下翻轉張量，返回一個新的張量。

向上/向下翻轉每列中的條目。行被保留，但以與以前不同的順序出現。

注意

要求張量至少是一維的。

注意

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

例子：

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