cupyx.scipy.linalg.lu_solve#

cupyx.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

Parameters:
  • lu_and_piv (tuple) – LU factorization of matrix a ((M, M)) together with pivot indices.

  • b (cupy.ndarray) – The matrix with dimension (M,) or (M, N).

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

    Type of system to solve:

    trans

    system

    0

    a x = b

    1

    a^T x = b

    2

    a^H x = b

  • overwrite_b (bool) – Allow overwriting data in b (may enhance performance)

  • check_finite (bool) – 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:

The matrix with dimension (M,) or (M, N).

Return type:

cupy.ndarray