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)[源代码][源代码]#
解方程 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' or '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)的矩阵。- 返回类型: