cupy.linalg.solve#

cupy.linalg.solve(a, b)[source]#

求解线性矩阵方程。

它计算 ax = b 中 x 的精确解，其中 a 是一个方阵且为满秩矩阵。

参数:

  • a (cupy.ndarray) – 维度为 (..., M, M) 的矩阵。

  • b (cupy.ndarray) – 维度为 (..., M) 或 (..., M, K) 的矩阵。

返回值:

维度为 (..., M) 或 (..., M, K) 的矩阵。

返回类型:

cupy.ndarray

警告

此函数调用一个或多个 cuSOLVER 例程，如果输入条件不满足，可能会产生无效结果。要检测这些无效结果，您可以在 cupyx.errstate() 或 cupyx.seterr() 中将 linalg 配置设置为非 ignore 的值。

另请参阅

numpy.linalg.solve()