pytorch tensor flip

因为使用中用到数据 需要flip，在python代码中使用numpy的flip很方便，但是c++中使用flip就不那么方便了。
本来图像数据，想要使用vtk去做数据处理，后来发现使用pytoch的torch处理更合适一些。

torch.flip(input,dim):第一个参数是输入，第二个参数是输入的第几维度，按照维度对输入进行翻转

# -*- coding : UTF-8 -*-
# @file   : torch_t1.py
# @Time   : 2021/8/29 22:37
# @Author : wmz
import torch
x = torch.arange(16).view(2, 2