快捷方式

torch.flip

torch.flip(input, dims) Tensor

沿 dims 中给定轴反转 n 维张量的顺序。

注意

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

参数
  • input (Tensor) – 输入张量。

  • dims (列表元组) – 要反转的轴

示例

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

        [[ 4,  5],
         [ 6,  7]]])
>>> torch.flip(x, [0, 1])
tensor([[[ 6,  7],
         [ 4,  5]],

        [[ 2,  3],
         [ 0,  1]]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源