jax.numpy.linalg.slogdet#

jax.numpy.linalg.slogdet(a, *, method=None)[来源]#

计算数组行列式的符号和(自然)对数。

numpy.linalg.slogdet() 的 JAX 实现。

参数:
  • a (ArrayLike) – 形状为 (..., M, M) 的数组,用于计算符号和对数行列式。

  • method (str | None) –

    用于行列式计算的方法。选项包括

    • 'lu'(默认):使用 LU 分解。

    • 'qr':使用 QR 分解。

返回:

数组元组 (sign, logabsdet),每个数组的形状为 a.shape[:-2]

  • sign 是行列式的符号。

  • logabsdet 是行列式绝对值的自然对数。

返回类型:

SlogdetResult

另请参阅

jax.numpy.linalg.det():直接计算行列式

示例

>>> a = jnp.array([[1, 2],
...                [3, 4]])
>>> sign, logabsdet = jnp.linalg.slogdet(a)
>>> sign  # -1 indicates negative determinant
Array(-1., dtype=float32)
>>> jnp.exp(logabsdet)  # Absolute value of determinant
Array(2., dtype=float32)