評價此頁

torch.flipud#

torch.flipud(input) Tensor#

沿上下方向翻轉張量,返回一個新張量。

沿上下方向翻轉每一列中的元素。行被保留,但出現的順序與之前不同。

注意

要求張量至少是 1-D。

注意

torch.flipud 會複製 input 的資料。這與 NumPy 的 np.flipud 不同,後者以恆定的時間返回一個檢視。由於複製張量的資料比檢視資料的工作量更大,因此 torch.flipud 的速度預計會比 np.flipud 慢。

引數

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

示例

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