Geometrically Constrained and Token-Based Probabilistic Spatial Transformers