jax.lax.linalg.tridiagonal_solve

jax.lax.linalg.tridiagonal_solve(dl, d, du, b)

計算三對角線性系統的解。

此函數計算三對角線性系統的解

\[A . X = B\]

參數:

• dl (Array) – 形狀為 [..., m] 的向量批次。A 的下對角線：dl[i] := A[i, i-1]，適用於 i 在 [0,m) 範圍內。請注意 dl[0] = 0。

• d (Array) – 形狀為 [..., m] 的向量批次。A 的中間對角線：d[i]  := A[i, i]，適用於 i 在 [0,m) 範圍內。

• du (Array) – 形狀為 [..., m] 的向量批次。A 的上對角線：du[i] := A[i, i+1]，適用於 i 在 [0,m) 範圍內。請注意 dl[m - 1] = 0。

• b (Array) – 右手邊矩陣。

回傳:

三對角系統的解 X。

回傳類型:

Array