jax.Array.diagonal

abstract Array.diagonal(offset=0, axis1=0, axis2=1)

Return the specified diagonal from the array.

Refer to jax.numpy.diagonal() for the full documentation.

Parameters:

• self (Array)

• offset (int)

• axis1 (int)

• axis2 (int)

Return type:

Array