cupyx.scipy.linalg.lu_factor#

cupyx.scipy.linalg.lu_factor(a, overwrite_a=False, check_finite=True)[source]#

LU 分解。

将给定的二维方阵分解为 P * L * U,其中 P 是置换矩阵,L 是单位对角元下三角矩阵,U 是上三角矩阵。

参数:
  • a (cupy.ndarray) – 维度为 (M, N) 的输入矩阵

  • overwrite_a (bool) – 是否允许覆盖 a 中的数据(可能会提高性能)

  • check_finite (bool) – 是否检查输入矩阵仅包含有限数。禁用此检查可能会获得性能提升,但如果输入包含无穷大或 NaN,则可能会导致问题(崩溃、无法终止)。

返回:

(lu, piv),其中 lu 是一个 cupy.ndarray,其上三角存储 U,下三角(不包含单位对角元)存储 Lpiv 是一个 cupy.ndarray,存储表示置换矩阵 P 的轴心索引。对于 0 <= i < min(M,N),矩阵的第 i 行与第 piv[i] 行进行了交换。

返回类型:

tuple