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