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
- 1
G. Strang, Linear Algebra and Its Applications, 2nd Ed., Orlando, FL, Academic Press, Inc., 1980, pg. 22.