mlx.core.linalg.lu_factor

lu_factor(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]

計算 LU 分解的緊湊表示。

參數:

• a (array) -- Input array.

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

回傳:

The LU matrix and pivots array.

回傳型別:

tuple(array, array)