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) >>> output = ops.Cholesky()(input_x) >>> print(output) [[1. 0.] [1. 1.]]