jax.lax.linalg.lu

jax.lax.linalg.lu(x)

带有部分主元的 LU 分解。

计算矩阵分解

\[P.A = L.U\]

其中 \(P\) 是 \(A\) 的行置换，\(L\) 是对角线元素为单位元的下三角矩阵，\(U\) 是上三角矩阵。

参数:

x (类似数组) – 形状为 [..., m, n] 的一批矩阵。

返回值:

一个元组 (lu, pivots, permutation)。

lu 是一个与 x 具有相同形状和数据类型的矩阵批次，其下三角包含 \(L\) 矩阵，上三角包含 \(U\) 矩阵。\(L\) 的（单位）对角线元素未显式表示。

pivots 是一个形状为 [..., min(m, n)] 的 int32 数组，表示应该对 \(A\) 执行的一系列行交换。

permutation 是行交换序列的另一种表示形式，表示为形状为 [..., m] 的 int32 数组。

返回类型:

tuple[Array, Array, Array]