Spatial Transformer Network with Mnist - View it on GitHub