mlx.core.linalg.solve_triangular

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

Computes the solution of a triangular system of linear equations AX = B.

參數:

• a (array) -- Input array.

• b (array) -- Input array.

• upper (bool, optional) -- Whether the array is upper or lower triangular. Default: False.

• 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