Added description
Extract the diagonal part of matrix as a matrix