Diag

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：将输入tensor(展平视为1D)的对角线元素，展开为2D对角矩阵

• 计算公式：

  设：

  $$\mathbf{x}\in {\mathbb{R}}^n：输入向量$$

  $$\mathbf{y}\in {\mathbb{R}}^{n\times n}：输出对角矩阵$$

  其中：

  $${\mathbf{y}}_{i,i}=x_i$$

  $${\mathbf{y}}_{i,j}=0当i\ne j$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	公式中的x。	INT64、INT32、FLOAT、FLOAT16、DOUBLE、BF16、COMPLEX64、COMPLEX128	ND
y	输出	公式中的y。	INT64、INT32、FLOAT、FLOAT16、DOUBLE、BF16、COMPLEX64、COMPLEX128	ND

• Atlas 训练系列产品、Atlas 推理系列产品、Atlas 200I/500 A2 推理产品、Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品：不支持BF16。

约束说明

只支持输入shape维度1-4维，对应输出2-8维，不支持标量输入。

调用说明

调用方式	调用样例	说明
图模式调用	NA	通过算子IR构图方式调用diag算子