jax.Array.trace

内容

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