mlx.core.linalg.lu_factor

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

计算 LU 分解的紧凑表示。

参数:

• a (array) – 输入数组。

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

返回:

LU 矩阵和 pivots 数组。

返回类型:

tuple(array, array)