Domanda di colloquio di Google Cloud

Simple matrix manipulation question !