Eye

产品支持情况

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

功能说明

算子功能：返回一个二维张量，该张量的对角线上元素值为1，其余元素值为0。

参数说明

参数名	输入/输出	描述	数据类型	数据格式
n	输入	表示第一个维度，Host侧的整型。取值范围不小于0。	-	-
m	输入	表示第二个维度，Host侧的整型。取值范围不小于0。	-	-
batch_shape	输入	可选，表示输出张量的batch维度，Host侧的整型列表，每项必须大于0。未指定时输出为二维。	-	-
dtype	输入	可选，表示输出张量的数据类型，Host侧的整型，对应数据类型枚举值。默认值为FLOAT32。	-	-
out	输出	表示输出张量，Device侧的aclTensor。shape为batch_shape + (n, m)，未指定batch_shape时为二维(n, m)。	BFLOAT16、FLOAT16、FLOAT32、INT32、INT16、INT8、UINT8、INT64、BOOL	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_eye	通过aclnnEye接口方式调用Eye算子。