文件最后提交记录最后更新时间
4 个月前
2 个月前
4 个月前
4 个月前
16 天前
README

Renorm

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
cc 哈尔滨工业大学-苏统华团队 Renorm 2025/12/22 新增Renorm算子

支持的产品型号

  • Atlas A2训练系列产品
  • Atlas 200I/500 A2推理产品

算子描述

  • 功能描述

    Renorm算子返回输入数据经过重归一化运算的结果。

  • 原型信息

    算子类型(OpType)Renorm
    nameTypedata typeformat
    算子输入
    xtensorfloat32,float16ND
    算子输出 ytensorfloat32,float16ND
    核函数名renorm

约束与限制

  • x,y,out的数据类型仅支持float32,float16,数据格式仅支持ND

算子使用

使用该算子前,请参考CANN开发套件包安装文档完成开发运行环境的部署。

编译部署

  • 进入到仓库目录

    cd ${git_clone_path}/ops-math
    
  • 执行编译

    bash build.sh --pkg --experimental --soc=ascend910b --ops=renorm_v2
    
  • 部署算子包

    ./build_out/cann-ops-<vendor_name>-linux.<arch>.run
    

算子调用

  • 执行调用

    bash build.sh --run_example renorm_v2 eager cust --vendor_name=custom
    

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_renorm_v2 通过aclnnRenormV2接口方式调用Renorm算子。