Anonymous interview question

Mini Attention matrix computation exercise