jax.numpy.linalg.solve#
- jax.numpy.linalg.solve(a, b)[source]#
Solve a linear matrix equation, or system of linear scalar equations.
LAX-backend implementation of
numpy.linalg.solve().Original docstring below.
Computes the “exact” solution, x, of the well-determined, i.e., full rank, linear matrix equation ax = b.
- Parameters:
a ((..., M, M) array_like) – Coefficient matrix.
b ({(..., M,), (..., M, K)}, array_like) – Ordinate or “dependent variable” values.
- Returns:
x – Solution to the system a x = b. Returned shape is identical to b.
- Return type:
{(…, M,), (…, M, K)} ndarray
References