嘗試動手實現一個 transformer2 — 多頭注意力機制