Sum of diagonal elements of a matrix
Example
A=
The size of the input array is:
X