pytorch version of spatial transformer networks - View it on GitHub