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 是一個形狀為 [..., min(m, n)] 的 int32 陣列，表示應在 \(A\) 上執行的一系列列交換。

permutation 是列交換序列的另一種表示形式，表示為形狀為 [..., m] 的 int32 陣列。

傳回類型:

tuple[Array, Array, Array]