mindspore.ops.Cholesky

class mindspore.ops.Cholesky(upper=False)[源代码]

计算单个或成批对称正定矩阵的Cholesky分解。

警告

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

更多参考详见 mindspore.ops.cholesky() 。

参数：

upper (bool，可选) - 为 True 返回上三角矩阵，否则返回下三角矩阵。默认值： False 。

输入：

input_x (Tensor) - shape大小为 \((*, N, N)\) ，其中 \(*\) 是零个或多个由对称正定矩阵组成的批处理维，数据类型为float32或float64。

输出：

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

支持平台：

GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([[1.0, 1.0], [1.0, 2.0]]), mindspore.float32)
>>> cholesky = ops.Cholesky(upper=False)
>>> output = cholesky(input_x)
>>> print(output)
[[1. 0.]
 [1. 1.]]