Rsqrt

产品支持情况

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

功能说明

• 算子功能：求输入张量x每个元素的平方根的倒数，并将结果保存到输出张量y中。

• 计算公式：

$$y=\frac{1}{\sqrt{x}}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量，公式中的x。	FLOAT、FLOAT16、BFLOAT16	ND
y	输出	输出张量，公式中的y，shape与x一致。	FLOAT、FLOAT16、BFLOAT16	ND

• Atlas 200I/500 A2推理产品、Atlas推理系列产品、Atlas训练系列产品：不支持BFLOAT16。

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_rsqrt	通过aclnnRsqrt或aclnnInplaceRsqrt接口方式调用Rsqrt算子。
图模式调用	test_geir_rsqrt	通过算子IR构图方式调用Rsqrt算子。