Cholesky
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | × |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | × |
| Atlas 训练系列产品 | × |
功能说明
-
算子功能：计算实数对称正定矩阵的Cholesky分解。
-
计算公式：
self=out×outHself = out \times out^{H}
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| self | 输入 | 表示传入的tensor，公式中的self。 | FLOAT、BFLOAT16 | ND |
| upper | 输入 | 表示选择上三角矩阵分解还是下三角矩阵分解。 | FLOAT、BFLOAT16 | ND |
| out | 输出 | 表示传出的tensor，公式中的out。 | FLOAT、BFLOAT16 | ND |
约束说明
- self的维度必须大于等于2，并且需为正定矩阵。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_linalg_cholesky | 通过aclnnLinalgCholesky接口方式调用Cholesky算子。 |