Pregunta de entrevista de General Motors (GM)

Code the multi-head attention mechanism