https://www.pylessons.com/transformer-attention