torch.flip¶
- torch.flip(input, dims) Tensor ¶
沿 dims 中给定轴反转 n 维张量的顺序。
注意
torch.flip 会复制
input
的数据。这与 NumPy 的 np.flip 不同,后者会以恒定时间返回视图。由于复制张量的数据比查看该数据的工作量更大,因此 torch.flip 预计比 np.flip 慢。示例
>>> 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]]])