Reference implementation: https://github.com/glassroom/sata_attention