NVIDIA interview question

Matrix manipulations