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 (类似数组) – 一批形状为 [..., m, m] 的矩阵。

• b (类似数组) – 如果 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 的矩阵。

返回类型:

数组