torch.flip(input, dims)详解

torch.flip(input, dims)是pytorch中的函数，用于对输入张量的指定维度进行翻转操作。

具体来说，torch.flip(input, dims)函数会返回一个新的张量，其中的元素按照指定维度进行翻转。翻转后的元素顺序与原始张量在指定维度上的顺序相反。

input是输入张量，dims是一个整数或整数列表，表示要进行翻转的维度。如果dims是一个整数，则只对该维度进行翻转；如果dims是一个整数列表，则对列表中的所有维度进行翻转。

下面是一个示例，

import torch

x = torch.tensor([[1, 2, 3],
                  [4, 5, 6],
                  [7, 8, 9]])

result1 = torch.flip(x, dims=0)
result2 = torch.flip(x, dims=[0,1])

print(result1)
print(result2)

"""
result1 = tensor([[7, 8, 9],
                  [4, 5, 6],
                  [1, 2, 3]])
result2 = tensor([[9, 8, 7],
                  [6, 5, 4],
                  [3, 2, 1]])
"""

注意，torch.flip(input, dims)函数不会改变原始张量，而是返回一个新的翻转后的张量。