mindspore.ops.cholesky_inverse

mindspore.ops.cholesky_inverse(input_x, upper=False)[源代码]

计算对称正定矩阵的逆矩阵。

如果 upper 为False，则返回的矩阵 \(U\) 为下三角矩阵，输出的结果：

\[inv = (UU^{T})^{-1}\]

如果 upper 为True，则返回的矩阵 \(U\) 为上三角矩阵，输出的结果：

\[inv = (U^{T}U)^{-1}\]

说明

输入Tensor必须是一个上三角矩阵或者下三角矩阵。

参数：

input_x (Tensor) - 输入Tensor，其rank为2，数据类型为float32或float64。
upper (bool) - 返回上三角矩阵还是下三角矩阵的标志。默认值：False。

返回：

Tensor，shape和数据类型与 input_x 相同。

异常：

TypeError - 如果 input_x 不是Tensor。
TypeError - 如果 input_x 的数据类型既不是float32，也不是float64。
ValueError - 如果 input_x 的维度不等于2。

支持平台：

GPU CPU

样例：

>>> input_x = Tensor(np.array([[2,0,0], [4,1,0], [-1,1,2]]), mindspore.float32)
>>> output = ops.cholesky_inverse(input_x)
>>> print(output)
[[ 5.8125 -2.625   0.625 ]
 [-2.625   1.25   -0.25  ]
 [ 0.625  -0.25    0.25  ]]