快捷方式

torch.flipud

torch.flipud(input) Tensor

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

沿上下方向翻转每列中的条目。行被保留,但出现的顺序与之前不同。

注意

要求张量至少为 1 维。

注意

torch.flipud 会复制 input 的数据。这与 NumPy 的 np.flipud 不同,后者在常数时间内返回一个视图。由于复制张量数据比查看数据需要更多工作,因此 torch.flipud 预计会比 np.flipud 慢。

参数

input (张量) – 必须至少为 1 维。

示例

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深度教程

查看教程

资源

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

查看资源