mlx.core.linalg.solve

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

計算線性方程組 AX = B 的解。

參數:

• a (array) -- 輸入陣列。

• b (array) -- 輸入陣列。

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

回傳值:

The unique solution to the system AX = B.

回傳值型別:

array