Compute Arithmetic Mean Along Matrix Axis - Coding problem | OneNoughtOne