https://singularityfeed.com/a-simple-implementation-of-the-attention-mechanism-from-scratch/