cupyx.scipy.linalg.lu_solve#
- cupyx.scipy.linalg.lu_solve(lu_and_piv, b, trans=0, overwrite_b=False, check_finite=True)[source]#
求解方程组
a * x = b
,已知矩阵a
的 LU 分解- 参数:
lu_and_piv (tuple) – 矩阵
a
((M, M)
) 的 LU 分解以及枢轴索引。b (cupy.ndarray) – 维度为
(M,)
或(M, N)
的矩阵。trans ({0, 1, 2}) –
要解决的系统类型
trans
系统
0
a x = b
1
a^T x = b
2
a^H x = b
overwrite_b (bool) – 是否允许覆盖 b 中的数据(可能提高性能)
check_finite (bool) – 是否检查输入矩阵只包含有限数。禁用此选项可能会提高性能,但如果输入包含无穷大或 NaN,则可能导致问题(崩溃、无法终止)。
- 返回:
维度为
(M,)
或(M, N)
的矩阵。- 返回类型: