jax.lax.linalg.triangular_solve

jax.lax.linalg.triangular_solve(a, b, *, left_side=False, lower=False, transpose_a=False, conjugate_a=False, unit_diagonal=False)

三角求解。

求解矩陣方程式

\[\mathit{op}(A) . X = B\]

如果 left_side 為 True 或

\[X . \mathit{op}(A) = B\]

如果 left_side 為 False。

A 必須是下或上三角方陣，其中 \(\mathit{op}(A)\) 可能會轉置 \(A\) （如果 transpose_a 為 True）和/或取其複共軛（如果 conjugate_a 為 True）。

參數：

• a (ArrayLike) – 形狀為 [..., m, m] 的矩陣批次。

• b (ArrayLike) – 如果 left_side 為 True，則形狀為 [..., m, n] 的矩陣批次；否則形狀為 [..., n, m]。

• left_side (bool) – 描述要解的兩個矩陣方程式中的哪一個；請參閱上方。

• lower (bool) – 描述應使用 a 的哪個三角形。另一個三角形會被忽略。

• transpose_a (bool) – 如果為 True，則會轉置 a 的值。

• conjugate_a (bool) – 如果為 True，則在求解中使用 a 的複共軛。如果 a 為實數，則無效。

• unit_diagonal (bool) – 如果為 True，則假設 a 的對角線為單位（全部為 1），且不會被存取。

傳回：

與 b 相同形狀和 dtype 的矩陣批次。

傳回型別：

Array