评价此页

torch.flipud#

torch.flipud(input) Tensor#

沿上/下方向翻转张量,返回一个新张量。

沿上/下方向翻转每列中的元素。行被保留,但顺序与之前不同。

注意

要求张量至少是一维的。

注意

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