评价此页

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