Parallel Attention Heads Mechanism - Coding problem | OneNoughtOne