CANN/ops-math Mod取模算子
Mod【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能返回 self 除以 other 的余数。计算公式对于入参 self 和比较标量 otherFmod 可以用如下数学公式表示$$ out_{i} self_{i} - (other \times trunc(self_{i}/other)) $$参数说明参数名输入/输出/属性描述数据类型数据格式self输入待进行mod计算的入参公式中的self_i。BFLOAT16、FLOAT16、FLOAT32、INT32NDother输入待进行mod计算的入参公式中的other。BFLOAT16、FLOAT16、FLOAT32、INT32NDout输出待进行mod计算的出参公式中的out_i。BFLOAT16、FLOAT16、FLOAT32、INT32ND约束说明aclnn 层支持 DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8 类型推导AICore kernel 覆盖 BFLOAT16、FLOAT16、FLOAT32、INT32其余类型走 AICPU fallback。self和out的shape必须一致。数据维度不支持8维以上。调用说明调用方式样例代码说明aclnn接口test_aclnn_fmod_scalar通过aclnnFmodScalar接口方式调用Mod算子。aclnn接口test_aclnn_fmod_tensor通过aclnnFmodTensor接口方式调用Mod算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

相关新闻