Attention DEC-heads