FilesLast commitLast update
17 days ago
14 days ago
3 months ago
1 month ago
11 days ago
3 days ago
1 month ago
17 days ago
11 days ago
3 days ago
29 days ago
9 days ago
README

Rsqrt

English | 简体中文

Product Support Status

Product Supported
Ascend 950PR&950DT products √
Atlas A3 products √
Atlas A2 products √
Atlas 200I/500 A2 inference products √
Atlas inference products √
Atlas training products √

Function Description

  • Operator function: computes the reciprocal of the square root of each element in the input tensor x, and stores the result in the output tensor y.

  • Computation formula:

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

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
x Input Input tensor, which is x in the formula. FLOAT, FLOAT16, BFLOAT16 ND
y Output Output tensor, which is y in the formula. Its shape is the same as that of x. FLOAT, FLOAT16, BFLOAT16 ND
  • Atlas 200I/500 A2 inference products, Atlas inference products, and Atlas training products: BFLOAT16 is not supported.

Constraint Description

None

Invocation Description

Invocation Method Invocation Sample Description
aclnn invocation test_aclnn_rsqrt Invokes the Rsqrt operator through the aclnnRsqrt or aclnnInplaceRsqrt API.
Graph-mode invocation test_geir_rsqrt Invokes the Rsqrt operator by building a graph through the operator IR.