Element 'multi-head attention'