mindspore.ops.cholesky

mindspore.ops.cholesky(input_x, upper=False)

计算给定对称正定矩阵的Cholesky分解。

如果 upper 为True，则返回的矩阵 $U$ 为上三角矩阵，分解形式为：

$$A=U^{T}U$$

如果 upper 为False，则返回的矩阵 $L$ 为下三角矩阵，分解形式为：

$$A=L{L}^T$$

参数：

• input_x (Tensor) - shape大小为 $(\ast ,N,N)$ 的输入tensor, 公式中的 $A$ 。

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

返回：

Tensor

支持平台：

GPU CPU

样例：

>>> import mindspore
>>> input_x = mindspore.tensor([[1.0, 1.0], [1.0, 2.0]])
>>> output = mindspore.ops.cholesky(input_x, upper=False)
>>> print(output)
[[1. 0.]
 [1. 1.]]