harmony 鸿蒙PowFusion
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后的形状决定。
起始版本： 3.2
相关模块：NNRt
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| float scale | 对x的数值进行缩放。 |
| float shift | 对x的数值缩放后的值进行增减。 |
类成员变量说明
scale
float PowFusion::scale
描述
对x的数值进行缩放。
shift
float PowFusion::shift
描述
对x的数值缩放后的值进行增减。
你可能感兴趣的鸿蒙文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦