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