DiagPart
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas 800I A2 推理产品 | √ |
功能说明
-
算子功能:提取输入矩阵的对角线元素。
-
计算公式:
yi=xi,iy_i = x_{i,i} yi=xi,i
对于输入形状为 [N, N] 的矩阵 x,输出形状为 [N] 的向量 y,其中 y[i] = x[i][i]。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待提取对角线元素的输入矩阵,形状必须为 [N, N] 的方阵。 | float16,float32,int32 | ND |
| y | 输出 | 提取的对角线元素,形状为 [N] 的向量。 | float16,float32,int32 | ND |
约束说明
- 输入张量 x 必须是二维方阵,形状为 [N, N]。
- 输入和输出的数据类型必须一致。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_diag_part | 通过aclnnDiagPart接口方式调用DiagPart算子。 |
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| fangfangssj | 个人开发者 | DiagPart | 2026/4/15 | DiagPart算子适配开源仓 |