jax.scipy.linalg.lu_solve

jax.scipy.linalg.lu_solve(lu_and_piv, b, trans=0, overwrite_b=False, check_finite=True)

Solve an equation system, a x = b, given the LU factorization of a

LAX-backend implementation of scipy.linalg._decomp_lu.lu_solve().

Does not support the Scipy argument check_finite=True, because compiled JAX code cannot perform checks of array values at runtime.

Does not support the Scipy argument overwrite_*=True.

Original docstring below.

Parameters:

• b (array) – Right-hand side

• trans ({0, 1, 2}, optional) –

  Type of system to solve:

  trans	system
  0	a x = b
  1	a^T x = b
  2	a^H x = b

• lu_and_piv (tuple[Array, Union[Array, ndarray, bool_, number, bool, int, float, complex]])

• overwrite_b (bool)

• check_finite (bool)

Returns:

x – Solution to the system

Return type:

array