mindspore.ops.matrix_solve

mindspore.ops.matrix_solve(matrix, rhs, adjoint=False)

求解线性方程组。

$$\begin{array}{r}\begin{array}{rl}& matrix[...,M,M]\ast x[...,M,K]=rhs[...,M,K]\\ & adjoint(matrix[...,M,M])\ast x[...,M,K]=rhs[...,M,K]\end{array}\end{array}$$

警告

• 当平台为GPU时，如果 matrix 中的矩阵不可逆，将产生错误或者返回一个未知结果。

参数：

• matrix (Tensor) - 第一个输入tensor。

• rhs (Tensor) - 第二个输入tensor。

• adjoint (bool，可选) - 表示是否需要在求解前对输入矩阵 matrix 做共轭转置，默认 False 。

返回：

Tensor

支持平台：

Ascend CPU

样例：

>>> import mindspore
>>> matrix = mindspore.tensor([[5., 4.], [3., 1.]])
>>> rhs = mindspore.tensor([[7.], [2.]])
>>> result = mindspore.ops.matrix_solve(matrix, rhs)
>>> print(result)
[[0.14285707]
 [1.5714287 ]]