mlx.core.linalg.lu_factor

Contents

mlx.core.linalg.lu_factor#

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

Computes a compact representation of the LU factorization.

Parameters:

  • 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.

Returns:

The LU matrix and pivots array.

Return type:

tuple(array, array)

Contents