Rsqrt
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| skywang2 | 个人开发者 | Rsqrt | 2026/03/30 | 新增Rsqrt算子 |
支持的产品型号
- Atlas A2训练系列产品
产品形态详细说明请参见昇腾产品形态说明
算子描述
-
功能描述
Rsqrt算子将数据进行开方并取倒数运算。 -
原型信息
算子类型(OpType) Rsqrt name Type data type format 算子输入 x tensor float32,float16,bfloat16 ND 算子输出 y tensor float32,float16,bfloat16 ND 核函数名 rsqrt
约束与限制
- x,y,out的数据类型只支持 float32,float16,bfloat16,数据格式只支持ND
运行验证
测试命令调用方式:build.sh
| 目录 | 描述 |
|---|---|
| test_aclnn_rsqrt.cpp | 通过aclnn调用的方式调用Rsqrt算子。 |