Rsqrt

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
skywang2 个人开发者 Rsqrt 2026/03/30 新增Rsqrt算子

支持的产品型号

  • Atlas A2训练系列产品

产品形态详细说明请参见昇腾产品形态说明

算子描述

  • 功能描述

    Rsqrt算子将数据进行开方并取倒数运算。

  • 原型信息

    算子类型(OpType)Rsqrt
    nameTypedata typeformat
    算子输入
    xtensorfloat32,float16,bfloat16ND
    算子输出 ytensorfloat32,float16,bfloat16ND
    核函数名rsqrt

约束与限制

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

运行验证

测试命令调用方式:build.sh

目录描述
test_aclnn_rsqrt.cpp通过aclnn调用的方式调用Rsqrt算子。