题目 ID: q-6011

手撕MHA，没想到面试真手撕这玩意，第一次写有点紧张写错了不少，但是面试官扫了一眼居然说差不多