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