jax.scipy.linalg.lu_solveΒΆ

jax.scipy.linalg.lu_solve(lu_and_piv, b, trans=0, overwrite_b=False, check_finite=True)[source]ΒΆ

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

LAX-backend implementation of lu_solve().

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

  • overwrite_b (bool, optional) – Whether to overwrite data in b (may increase performance)

  • check_finite (bool, optional) – Whether to check that the input matrices contain only finite numbers. Disabling may give a performance gain, but may result in problems (crashes, non-termination) if the inputs do contain infinities or NaNs.

Returns

x – Solution to the system

Return type

array