cupyx.scipy.linalg.solve_triangular#
- cupyx.scipy.linalg.solve_triangular(a, b, trans=0, lower=False, unit_diagonal=False, overwrite_b=False, check_finite=False)[source]#
求解方程 a x = b 中的 x，假设 a 是一个三角矩阵。
- 参数:
a (cupy.ndarray) – 维度为
(M, M)的矩阵。b (cupy.ndarray) – 维度为
(M,)或(M, N)的矩阵。lower (bool) – 只使用
a的下三角部分包含的数据。默认使用上三角。trans (0, 1, 2, 'N', 'T' 或 'C') –
要解决的系统类型
’0’ 或 ‘N’ – \(a x = b\)
’1’ 或 ‘T’ – \(a^T x = b\)
’2’ 或 ‘C’ – \(a^H x = b\)
unit_diagonal (bool) – 如果为
True，则假定a的对角线元素为 1，并且不会引用它们。overwrite_b (bool) – 允许覆盖 b 中的数据（可能会提高性能）。
check_finite (bool) – 是否检查输入矩阵只包含有限数。禁用此选项可能会提高性能，但如果输入确实包含无穷大或 NaN，则可能会导致问题（崩溃、非终止）。
- 返回:
维度为
(M,)或(M, N)的矩阵。- 返回类型: