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) 的矩阵。

返回类型:

cupy.ndarray