CANN ops-math 幂运算算子
2026/5/11 7:47:02 网站建设 项目流程

Pows

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

产品支持情况

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

功能说明

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

$$ out_i = self_i^{exponent_i} $$

参数说明

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

约束说明

调用说明

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

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询