cupyx.scipy.linalg.lu_factor#

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

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 ,在下三角存储不带单位对角元素的 L ,而 piv 是一个 cupy.ndarray ,存储表示置换矩阵 P 的枢轴索引。对于 0 <= i < min(M,N) ,矩阵的第 i 行与第 piv[i] 行互换。

返回类型:

tuple