Thanks for the links.
I would like to ask if there any example to depict the combination of the the affine_grid and grid_sample along with the localization network (which is essentially a set of conv layers ending up to regress the 6 parameters of the transformation)?