Pergunta de entrevista da empresa Google Cloud

Simple matrix manipulation question !