mlx.core.linalg.solve

Contents

mlx.core.linalg.solve#

solve(a: array, b: array, *, stream: None | Stream | Device = None) → array#

Compute the solution to a system of linear equations AX = B.

Parameters:

  • a (array) – Input array.

  • b (array) – Input array.

  • stream (Stream, optional) – Stream or device. Defaults to None in which case the default stream of the default device is used.

Returns:

The unique solution to the system AX = B.

Return type:

array

Contents