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_side
是True
或\[ \begin{align}\begin{aligned}X . \mathit{op}(A) = B\\X . \mathit{op}(A) = B\end{aligned}\end{align} \]如果
left_side
是False
。A
必须是一个下三角或上三角的方阵,其中 \(\mathit{op}(A)\) 可能会在transpose_a
为True
时转置 \(A\),或者在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
形状和数据类型相同的矩阵。- 返回类型: