Function diag

Extract a diagonal or construct a diagonal array.

If v is a 2D array, return the k-th diagonal of v (as a view). If v is a 1D array, return a 2D array with v on the k-th diagonal.