Xdivy

产品支持情况

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

功能说明

  • 算子功能:计算x / y,当x == 0时结果为0。

  • 计算公式:

    out={0if x=0xyotherwiseout = \begin{cases} 0 & \text{if } x = 0 \\ \frac{x}{y} & \text{otherwise} \end{cases}

  • 支持broadcast:x和y的shape需满足broadcast关系,输出shape为广播后的最大值shape。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x1 输入 公式中的x,被除数。shape需与x2满足broadcast关系。 FLOAT、FLOAT16、BFLOAT16 ND
x2 输入 公式中的y,除数。shape需与x1满足broadcast关系。 FLOAT、FLOAT16、BFLOAT16 ND
y 输出 公式中的out,计算结果。shape为x1与x2 broadcast后的最大值shape。 FLOAT、FLOAT16、BFLOAT16 ND

约束说明

  • 输入shape维度最大为8。
  • 输入x1、x2、输出y的数据类型必须一致。

调用说明

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