dask.array.linalg.solve

dask.array.linalg.solve(a, b, sym_pos=False)[source]

Solve the equation a x = b for x. By default, use LU decomposition and forward / backward substitutions. When sym_pos is True, use Cholesky decomposition.

Parameters
a(M, M) array_like

A square matrix.

b(M,) or (M, N) array_like

Right-hand side matrix in a x = b.

sym_posbool

Assume a is symmetric and positive definite. If True, use Cholesky decomposition.

Returns
x(M,) or (M, N) Array

Solution to the system a x = b. Shape of the return matches the shape of b.