diag function

Matrix diagonals