DiagPart

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品	√

功能说明

• 算子功能：提取输入矩阵的对角线元素。

• 计算公式：

$$y_i=x_{i,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算子适配开源仓