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. |