Pow
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
-
接口功能：exponent每个元素作为input对应元素的幂完成计算。
-
计算公式：
outi=inputiexponentiout_i = input_i^{exponent_i}
参数说明
| 参数名 | 输入/输出 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| input | 输入 | 表示底数。数据类型与exponent满足TensorScalar互推导关系。 | FLOAT、FLOAT16、DOUBLE、INT16、INT32、INT64、INT8、UINT8、COMPLEX64、COMPLEX128、BFLOAT16 | ND |
| exponent | 输入 | 表示指数。数据类型与input满足TensorScalar互推导关系。 | FLOAT、FLOAT16、DOUBLE、INT16、INT32、INT64、INT8、UINT8、COMPLEX64、COMPLEX128、BFLOAT16 | ND |
| out | 输出 | 表示input的exponent次幂。数据类型需要是input的数据类型与exponent的数据类型推导之后可转换的数据类型（参见互转换关系）。 | FLOAT、FLOAT16、DOUBLE、INT16、INT32、INT64、INT8、UINT8、COMPLEX64、COMPLEX128、BFLOAT16 | ND |
约束说明
算子约束：INT32整型计算在如下范围以外的场景，会出现超时。
| shape | exponent_value |
|---|---|
| <=100000（十万） | -200000000~200000000(两亿) |
| <=1000000（百万） | -20000000~20000000(两千万) |
| <=10000000（千万） | -2000000~2000000(两百万) |
| <=100000000（亿） | -200000~200000(二十万) |
| <=1000000000（十亿） | -20000~20000(两万) |
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_exp2 | 通过aclnnExp2接口方式调用Pow算子。 |
| aclnn调用 | test_aclnn_pow_tensor_tensor | 通过aclnnPowTensorTensor接口方式调用Pow算子。 |
| 图模式调用 | test_geir_pow | 通过算子IR构图方式调用Pow算子。 |