jax.numpy.linalg.det

jax.numpy.linalg.det(a)

计算数组的行列式。

JAX 实现了 numpy.linalg.det()。

参数:

a (ArrayLike) – 用于计算行列式的形状为 (..., M, M) 的数组。

返回值:

形状为 a.shape[:-2] 的行列式数组。

返回类型:

数组

另请参阅

jax.scipy.linalg.det()：用于计算行列式的 SciPy 风格的 API。

示例

>>> a = jnp.array([[1, 2],
...                [3, 4]])
>>> jnp.linalg.det(a)
Array(-2., dtype=float32)