Tiny-Net: learned token embeddings in 2D