DataCompare

产品支持情况

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

功能说明

• 算子功能：逐元素比较两个相同shape和dtype的输入张量x1和x2，统计差异超出容差范围的元素总个数（All Reduce），输出FLOAT标量表示不匹配元素总数。

• 计算公式：

  逐元素判断是否不匹配：

  $$dif{f}_i=|x{1}_i-x{2}_i|$$

  $$threshol{d}_i=atol+rtol\times |x{2}_i|$$

  $$mismatc{h}_i=\{\begin{array}{ll}1.0,& dif{f}_i>threshol{d}_i\\ 0.0,& dif{f}_i\le threshol{d}_i\end{array}$$

  归约输出（不匹配元素总数）：

  $$output=\sum _{i}mismatc{h}_i(\text{dtype: FLOAT})$$

• 输出为0：所有元素均在容差范围内，两个张量视为匹配。

• 输出> 0：存在差异超差的元素，值越大说明差异越大。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x1	输入	第一个输入张量，对应公式中的x1。shape和dtype必须与x2完全一致，不支持broadcast。支持0-8维张量，支持空Tensor。	FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、INT32	ND
x2	输入	第二个输入张量，对应公式中的x2。shape和dtype必须与x1完全一致，不支持broadcast。支持0-8维张量，支持空Tensor。	FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、INT32	ND
atol	属性	绝对容差，对应公式中的atol。 需>=0。 默认值为1e-5。	FLOAT	-
rtol	属性	相对容差，对应公式中的rtol。 需>=0。 默认值为1e-3。	FLOAT	-
num	输出	不匹配元素总数，对应公式中的output。固定为FLOAT标量（0维张量）。	FLOAT	ND

约束说明

• x1和x2的shape必须完全相同，不支持broadcast。
• x1和x2的dtype必须完全相同，不支持混合dtype。
• 整数类型（INT8/UINT8/INT32）输入时，算子内部先cast到FLOAT再执行比较计算。
• 归约方式为All Reduce（对所有轴归约），用户不可指定axis。
• 输出固定为FLOAT标量。
• 确定性实现：相同输入保证产生相同输出。
• 当输入元素总数超过$2^{24}$（约16M）时，由于FLOAT精确表示范围限制，计数值可能不精确。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_data_compare	通过算子IR构图方式调用DataCompare算子。