Sqrt
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 square root of non-negative numbers, and returns nan for negative numbers.
-
Computation formula:
out=sqrt(self)={self,self≥0,nan,self<0out=sqrt(self)=\begin{cases} \sqrt {self}, & self\ge 0 , \\ nan, & self\lt 0 \end{cases}
Parameter Description
| Parameter Name | Input/Output/Attribute | Description | Data Type | Data Format |
|---|---|---|---|---|
| self | Input | Input parameter for the sqrt computation, which is self in the formula. | COMPLEX64, COMPLEX128, FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, BOOL | ND |
| out | Output | Output parameter for the sqrt computation, which is out in the formula. | FLOAT, FLOAT16, DOUBLE, COMPLEX64, COMPLEX128, BFLOAT16 | ND |
- Atlas training products and Atlas inference products: BFLOAT16 is not supported.
Constraint Description
None
Invocation Description
| Invocation Method | Invocation Sample | Description |
|---|---|---|
| aclnn invocation | test_aclnn_sqrt | Invokes the Sqrt operator through the aclnnSqrt API. |
| Graph-mode invocation | test_geir_sqrt | Invokes the Sqrt operator by building a graph through the operator IR. |