Spatial Transformer Layer