mm.math.set_diag

math.set_diag(diag, k)

The array with the diagonal set to diag.

Parameters:

• array (ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]) – The array to set the diagonal of

• diag (ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]) – The diagonal to set

• k (int) – diagonal to set

Return type:

ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]

Returns:

The array with the diagonal set to diag

code/api/mrmustard.math.set_diag

 

Download Python script

 

Download Notebook

 

View on GitHub