jax.Array.diagonal

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

從陣列傳回指定的對角線。

請參閱 jax.numpy.diagonal() 以取得完整文件。

參數:

• self (Array)

• offset (int)

• axis1 (int)

• axis2 (int)

傳回型別:

Array