jax.Array.trace# abstract Array.trace(offset=0, axis1=0, axis2=1, dtype=None, out=None)[source]# 返回沿对角线的总和。 有关完整文档,请参阅 jax.numpy.trace()。 参数: self (Array) offset (int | ArrayLike) axis1 (int) axis2 (int) dtype (DTypeLike | None) out (None) 返回类型: Array