Scaled Dot-Product Attention

Implement the scaled dot-product attention mechanism from the Transformer architecture.