AdjacentDifference

产品支持情况

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

功能说明

  • 算子功能:比较输入张量相邻元素的差异,若相邻元素相同,返回0,否则返回1。

  • 计算公式:

outi=(xi==xj? 0:1).to(ydtype) j=i−1out_i = {(x_i == x_j ?\ 0 : 1)}.to(y_dtype) \ j = i - 1

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待进行AdjacentDifference计算的入参,公式中的x。 FLOAT、FLOAT16、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
y_dtype 输入 属性,指定输出类型,公式中的y_dtype。 INT32 ND
y 输出 待进行AdjacentDifference计算的出参,公式中的out。 INT32、INT64 ND
  • Atlas 训练系列产品、Atlas 推理系列产品: 不支持BFLOAT16。

约束说明

  • 无。

调用说明

调用方式 样例代码 说明
图模式 test_geir_adjacent_difference 通过算子IR构图方式调用AdjacentDifference算子。