jax.lax.linalg.lu

jax.lax.linalg.lu(x)

部分枢轴旋转的LU分解。

计算矩阵分解：

\[P.A = L.U\]

其中 \(P\) 是矩阵 \(A\) 的行排列，\(L\) 是一个对角元素为1的下三角矩阵，而 \(U\) 是一个上三角矩阵。

参数:

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

返回:

一个元组 (lu, pivots, permutation)。 lu 是一个与 x 形状和数据类型相同的矩阵批次，其中包含 \(L\) 矩阵在其下三角部分和 \(U\) 矩阵在其上三角部分。\(L\) 的（单位）对角元素没有显式表示。 pivots 是一个形状为 [..., min(m, n)] 的 int32 数组，表示应在对 \(A\) 执行的一系列行交换。 permutation 是行交换序列的另一种表示形式，作为排列，表示为形状为 [..., m] 的 int32 数组。

返回类型:

tuple[Array, Array, Array]