jax.Array.diagonal — JAX documentation (original) (raw)

Contents

• Array.diagonal()

jax.Array.diagonal#

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

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