cupynumeric.linalg.solve#

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

Single matrix multi-GPU usage is limited to cusolverMP. Additional multi-GPU/CPU usage is limited to data parallel matrix-wise batching.