Add multi head attention#21

zhenyu-02 wants to merge 1 commit intotekaratzas:maintekaratzas/RustGPT:mainfrom

zhenyu-02:feature/multi-head-attentionzhenyu-02/RustGPT:feature/multi-head-attentionCopy head branch name to clipboard