PowFusion

概述

求x的y次幂，输入必须是两个tensor或一个tensor和一个标量。

当输入是两个tensor时，它们的数据类型不能同时为DATA_TYPE_BOOL，且要求两个tensor的shape相同。当输入是一个tensor和一个标量时，标量只能是一个常数。

该算子对应的NodeType为NODE_TYPE_POW_FUSION。

参数说明：x的每个元素会做如下运算：x' = scale*x+shift，然后对于x'再求y次幂。

输入：

• x，实数、bool值或tensor，tensor的数据类型为实数或DATA_TYPE_BOOL。

• y，实数、bool值或tensor，tensor的数据类型为实数或DATA_TYPE_BOOL。

输出：

• output，形状由x和y broadcast后的形状决定。

Since:

3.2

Version:

1.0

相关模块:

NNRt

汇总

Public 属性

类成员变量说明

scale

float PowFusion::scale

描述:

对x的数值进行缩放。

shift

float PowFusion::shift

描述:

对x的数值缩放后的值进行增减。