General Motors (GM) Interview Question

Code the multi-head attention mechanism