jax.Array.diagonal

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

从数组中返回指定的对角线。

请参阅 jax.numpy.diagonal() 获取完整文档。

参数:

• self (Array)

• offset (int)

• axis1 (int)

• axis2 (int)

返回类型:

Array