diag

Create or extract diagonal

If input is 1D, creates a diagonal matrix. If input is 2D, extracts the diagonal elements.

Parameters

• x (NDArray): The input array.

Returns

The diagonal matrix or vector.

See Also

matmul