Is there any Spatial Transformer Layer kind of a thing in pytorch? I could find TransformerLayer in Lasagne which is the STN layer implementation.
EDIT 1:
If there is any example of STN with affine_grid and grid_sample as mentioned below, it would be of great help.