TruncateMod

产品支持情况

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

功能说明

  • 算子功能:返回self除以other的余数。

  • 计算公式:

    将张量self,和标量或张量other,进行广播成相同shape的张量后,TruncateMod可以用如下数学公式表示:

    outi=selfi−trunc(selfiotheri)∗otheriout_{i} = self_{i} - trunc \left(\frac{self_{i}}{other_{i}}\right) * other_{i}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
self 输入 待进行TruncateMod计算的入参,公式中的self_i。 DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8 ND
other 输入 待进行TruncateMod计算的入参,公式中的other_i。 DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8 ND
out 输出 待进行TruncateMod计算的出参,公式中的out_i。 DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8 ND

约束说明

  1. 数据类型需满足数据类型推导规则,推导后的数据类型需在支持的数据类型范围内。
  2. self和other的shape必须满足 广播规则
  3. 数据维度不支持8维以上。

调用说明

调用方式 样例代码 说明
geir接口 test_geir_truncate_mod 通过geir接口方式调用TruncateMod算子。