手撕 Self-Attention

题目描述

实现 Scaled Dot-Product Attention 机制。

思路分析

Q、K、V 三个矩阵，计算 softmax(QK^T / sqrt(d_k)) * V。