FloorMod

产品支持情况

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

功能说明

  • 算子功能:对输入张量进行逐元素取模运算,结果的符号与除数(x2)一致(向下取整)。

  • 计算公式:

yi=x1i−⌊x1ix2i⌋×x2iy_i = x1_i - \lfloor \frac{x1_i}{x2_i} \rfloor \times x2_i

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x1 输入 待进行FloorMod计算的被除数入参。 FLOAT、FLOAT16、BFLOAT16、INT32 ND
x2 输入 待进行FloorMod计算的除数入参。 FLOAT、FLOAT16、BFLOAT16、INT32 ND
y 输出 FloorMod计算的输出结果。 FLOAT、FLOAT16、BFLOAT16、INT32 ND

约束说明

  • x1x2 的 shape 需要满足广播(Broadcast)规则。
  • 数据类型需要一致,或者满足隐式类型转换规则。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_floor_mod.cpp 通过aclnnRemainder接口方式调用FloorMod算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
Tream 个人开发者 FloorMod 2025/12/17 FloorMod算子适配开源仓