jax.lax.linalg.三角形求解

jax.lax.linalg.三角形求解#

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_sideTrue

\[ \begin{align}\begin{aligned}X . \mathit{op}(A) = B\\X . \mathit{op}(A) = B\end{aligned}\end{align} \]

如果 left_sideFalse

A 必须是一个下三角或上三角的方阵,其中 \(\mathit{op}(A)\) 可能会在 transpose_aTrue 时转置 \(A\),或者在 conjugate_aTrue 时取其复共轭。

参数:
  • a (ArrayLike) – 一批形状为 [..., m, m] 的矩阵。

  • b (ArrayLike) – 如果 left_sideTrue,则是一批形状为 [..., 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 形状和数据类型相同的矩阵。

返回类型:

Array