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

样例:

>>> 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.]]