jax.numpy.linalg.solve

jax.numpy.linalg.solve(a, b)

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.