代码题：实现 Casual mask 的 MHA，并说明其计算复杂度。

回答思路

1. 正确实现因果掩码（下三角掩码）
2. 说明掩码处理位置（softmax 前加负无穷）
3. 分析注意力计算的时间与空间复杂度
4. 指出 QKV 投影、score、softmax、加权求和的复杂度构成