jax.Array.trace

abstract Array.trace(offset=0, axis1=0, axis2=1, dtype=None, out=None)

傳回沿對角線的總和。

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

參數:

• self (Array)

• offset (int | ArrayLike)

• axis1 (int)

• axis2 (int)

• dtype (DTypeLike | None)

• out (None)

傳回類型:

Array