Pows

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品
Atlas 推理系列产品
Atlas 训练系列产品
Kirin X90 处理器系列产品
Kirin 9030 处理器系列产品

功能说明

  • 算子功能:对input中的每个元素应用指数为exponent的幂运算。
  • 计算公式:

outi=selfiexponentiout_i = self_i^{exponent_i}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
self 输入 公式中的输入`self`,Device侧的aclTensor。 FLOAT、FLOAT16、BFLOAT16 ND
exponent 输入 公式中的输入`exponent`,Device侧的aclScalar。 FLOAT、FLOAT16、BFLOAT16 ND
out 输出 公式中的输出`out`,Device侧的aclTensor。 FLOAT、FLOAT16、BFLOAT16 ND
  • Kirin X90/Kirin 9030 处理器系列产品、Atlas 训练系列产品、Atlas 推理系列产品:不支持BFLOAT16。

约束说明

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_pows 通过算子IR构图方式调用Pows算子。