torch.flipud

torch.flipud(input) Tensor

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

沿上下方向翻转每列的条目。行保持不变,但顺序会有所不同。

注意

要求张量必须是一维或以上。

注意

torch.flipud 会复制 input 的数据。这与 NumPy 的 np.flipud 不同,np.flipud 返回的是常量时间内的视图。由于复制张量的数据比查看该数据的工作量大,因此预计torch.flipud 会比 np.flipud 更慢。

参数

输入 (Tensor) – 需要是至少一维的张量。

示例:

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