mindspore.ops.cholesky_solve

mindspore.ops.cholesky_solve(input, input2, upper=False)

根据Cholesky分解因子 input2 计算一组具有正定矩阵的线性方程组的解。

如果 upper 为 True， input2 是上三角矩阵，输出的结果：

$$output=(input{2}^T\ast input2{)}^{-1}input$$

如果 upper 为 False， input2 是下三角矩阵，输出的结果：

$$output=(input2\ast input{2}^T{)}^{-1}input$$

警告

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

参数：

• input (Tensor) - shape为 $(\ast ,N,M)$ 的输入tensor。

• input2 (Tensor) - shape为 $(\ast ,N,N)$ 的tensor，Cholesky因子。

• upper (bool, 可选) - 是否视为上三角矩阵。默认 False 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input1 = mindspore.tensor([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])
>>> input2 = mindspore.tensor([[2., 0., 0.], [4., 1., 0.], [-1., 1., 2.]])
>>> out = mindspore.ops.cholesky_solve(input1, input2, upper=False)
>>> print(out)
[[ 5.8125 -2.625   0.625 ]
 [-2.625   1.25   -0.25  ]
 [ 0.625  -0.25    0.25  ]]