jax.scipy.linalg.solve

jax.scipy.linalg.solve(a, b, sym_pos=False, lower=False, overwrite_a=False, overwrite_b=False, debug=False, check_finite=True)[source]

Solves the linear equation set a * x = b for the unknown x

LAX-backend implementation of solve().

Original docstring below.

for square a matrix.

If the data matrix is known to be a particular type then supplying the corresponding string to assume_a key chooses the dedicated solver. The available options are

generic matrix

‘gen’

symmetric

‘sym’

hermitian

‘her’

positive definite

‘pos’

If omitted, 'gen' is the default structure.

The datatype of the arrays define which solver is called regardless of the values. In other words, even when the complex array entries have precisely zero imaginary parts, the complex solver will be called based on the data type of the array.

Parameters

  • a ((N, N) array_like) – Square input data

  • b ((N, NRHS) array_like) – Input data for the right hand side.

  • sym_pos (bool, optional) – Assume a is symmetric and positive definite. This key is deprecated and assume_a = ‘pos’ keyword is recommended instead. The functionality is the same. It will be removed in the future.

  • lower (bool, optional) – If True, only the data contained in the lower triangle of a. Default is to use upper triangle. (ignored for 'gen')

  • overwrite_a (bool, optional) – Allow overwriting data in a (may enhance performance). Default is False.

  • overwrite_b (bool, optional) – Allow overwriting data in b (may enhance performance). Default is False.

  • 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 – The solution array.

Return type

(N, NRHS) ndarray