Eye

产品支持情况

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

功能说明

• 算子功能：创建一个二维矩阵，其中对角线上的值为1，其余值为0。
• 计算公式：

$$y_{ij}=\{\begin{array}{ll}1,& \text{if }i=j\\ 0,& \text{if }i\ne j\end{array}$$

参数说明

参数名	输入/输出	描述	数据类型	数据格式
x	输入	输入任意矩阵	见下方	ND
y	输出	输出的单位矩阵	见下方	ND

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：数据类型支持FLOAT、FLOAT16、INT32、INT16、BFLOAT16。

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_eye	通过[aclnnEye]接口方式调用eye算子。