mindspore.ops.lu_solve

mindspore.ops.lu_solve(b, LU_data, LU_pivots)

计算线性方程组 $Ay=b$ 的LU解。

说明

• b 的shape为 $(\ast ,m,k)$ ， LU_data 的shape为 $(\ast ,m,m)$ ， LU_pivots 的shape为 $(\ast ,m)$ ，$\ast$ 表示batch数量。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• b (Tensor) - 上面公式的列向量 b 。

• LU_data (Tensor) - LU分解的结果，上面公式中的 A 。

• LU_pivots (Tensor) - LU分解的主元，主元可以被转为变换矩阵P。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> b = mindspore.tensor([[1.], [3.], [3.]])
>>> LU_data = mindspore.tensor([[2., 1., 1.], [0.5, 1., 1.5], [0.5, 0., 2.5]])
>>> LU_pivots = mindspore.tensor(([2, 2, 3]), mindspore.int32)
>>> y = mindspore.ops.lu_solve(b, LU_data, LU_pivots)
>>> print(y)
[[ 1.9000001]
 [-1.4000001]
 [ 0.6      ]]