dpnp.linalg.solve

dpnp.linalg.solve(a, b)[source]

Solve a linear matrix equation, or system of linear scalar equations.

For full documentation refer to numpy.linalg.solve.

Parameters:

  • a ((..., M, M) {dpnp.ndarray, usm_ndarray}) -- Coefficient matrix.

  • b ({(…, M,), (…, M, K)} {dpnp.ndarray, usm_ndarray}) -- Ordinate or "dependent variable" values.

Returns:

out -- Solution to the system ax = b. Returned shape is identical to b.

Return type:

{(…, M,), (…, M, K)} dpnp.ndarray

See also

dpnp.dot

Returns the dot product of two arrays.

Examples

>>> import dpnp as dp
>>> a = dp.array([[1, 2], [3, 5]])
>>> b = dp.array([1, 2])
>>> x = dp.linalg.solve(a, b)
>>> x
array([-1.,  1.])

Check that the solution is correct:

>>> dp.allclose(dp.dot(a, x), b)
array([ True])