jax.Array.diagonal

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

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

有关完整文档，请参阅 jax.numpy.diagonal()。

参数：

• self (Array)

• offset (int)

• axis1 (int)

• axis2 (int)

返回类型：

Array