jax.lax.linalg.lu

jax.lax.linalg.lu(x)

使用部分旋转的 LU 分解。

计算矩阵分解

\[P.A = L.U\]

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

参数:

x (ArrayLike) – 一批形状为 [..., m, n] 的矩阵。

返回:

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

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

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

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

返回类型:

tuple[Array, Array, Array]