cupy.linalg.solve

cupy.linalg.solve(a, b)

求解线性矩阵方程。

它计算 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()