General Motors (GM) interview question

Code the multi-head attention mechanism