spatial_demo_attention.py