Pow

产品支持情况

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

功能说明

• 接口功能：exponent每个元素作为input对应元素的幂完成计算。

• 计算公式：

$$ou{t}_i=inpu{t}_i^{exponen{t}_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算子。