mlx.core.linalg.solve_triangular

mlx.core.linalg.solve_triangular#

solve_triangular(a: array, b: array, *, upper: bool = False, stream: None | Stream | Device = None) array#

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

参数:
  • a (array) – 输入数组。

  • b (array) – 输入数组。

  • upper (bool, 可选) – 数组是上三角还是下三角。默认值: False

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

返回:

方程组 AX = B 的唯一解。

返回类型:

array