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 是一批矩阵，其形状和 dtype 与 x 相同，其中包含下三角中的 \(L\) 矩阵和上三角中的 \(U\) 矩阵。 \(L\) 的（单位）对角线元素未明确表示。

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

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

返回类型:

tuple[Array, Array, Array]