mlx.core.linalg.solve

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

计算线性方程组 AX = B 的解。

参数:

• a (array) – 输入数组。

• b (array) – 输入数组。

• stream (Stream, 可选) – 流或设备。默认为 None，在这种情况下使用默认设备的默认流。

返回:

系统 AX = B 的唯一解。

返回类型:

array