TikTok interview question

Question about how to do some matrix operations