cupyx.scipy.linalg.lu_solve#

cupyx.scipy.linalg.lu_solve(lu_and_piv, b, trans=0, overwrite_b=False, check_finite=True)[源代码][源代码]#

给定矩阵 a 的 LU 分解,解方程组 a * x = b

参数:
  • 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) 的矩阵。

返回类型:

cupy.ndarray