cunumeric.linalg.solve#

cunumeric.linalg.solve(a: ndarray, b: ndarray, out: ndarray | None = None) → ndarray#

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

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.
out ({(M,), (M, K)}, array_like, optional) – An optional output array for the solution

Returns:

x – Solution to the system a x = b. Returned shape is identical to b.

Return type:

{(M,), (M, K)} ndarray

Raises:

LinAlgError – If a is singular or not square.

Notes

Multi-GPU usage is only available when compiled with cusolverMP.