pytorch的STN（空间变换网络）
STN可以直接插入现有的CNN中。
Localisation Network：一CNN生成变换参数。
Grid Genator：生成与输出图像的每个像素相对应的坐标网络。
Sampler：输出结果
#定义一个stn网络：本地网络（卷积池化relu卷积池化relu)Linear relu Linear得到3 * 2的仿射矩阵
#grid = F.affine_grid(theta, x.size())这个函数得到grid ,x = F.grid_sample(x, grid)得到stn的输出
利用stn将transform the input。