Diagonal of a matrix