jax.Array.diagonal#

abstractArray.diagonal(offset=0,axis1=0,axis2=1)[source]#

Return the specified diagonal from the array.

Refer tojax.numpy.diagonal() for the full documentation.

Parameters:

• self (Array)

• offset (int)

• axis1 (int)

• axis2 (int)

Return type:

Array