SWAT: Spatial Structure Within and Among Tokens. - Kahatapitiya, Kumara; Ryoo, Michael S. (CoRR, 2021)