harmony 鸿蒙ReduceFusion
ReduceFusion
概述
减小x张量的维度。
该算子对应的NodeType为NODE_TYPE_REDUCE_FUSION。
参数说明：mode若为REDUCE_ALL,REDUCE_PROD和REDUCE_MEAN，则reduce_to_end为true时，Reduce之后的输出乘以coeff为最终输出。
输入：
x，n维tensor，n<8。
axis，1维tensor，指定reduce的维度，axis中每个元素的取值范围为[-n，n)。
输出：
- output，执行Reduce之后的m维的tensor，其数据类型和x相同。当keepDims为false时，m<n；当keepDims为true时，m==n。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| keepDims | boolean 维度是否保持不变。 |
| mode | enum ReduceMode 减小张量维度的算法，详情请参考：ReduceMode。 |
| reduceToEnd | boolean 如果为true，则从axis取第一个元素并设置为i，然后axis会被修改为[i,i+1,…,n-1,n]。 |
| coeff | float 系数 |
类成员变量说明
coeff
float ReduceFusion::coeff
描述:
系数
keepDims
boolean ReduceFusion::keepDims
描述:
维度是否保持不变。
mode
enum ReduceMode ReduceFusion::mode
描述:
减小张量维度的算法，详情请参考：ReduceMode。
reduceToEnd
boolean ReduceFusion::reduceToEnd
描述:
如果为true，则从axis取第一个元素并设置为i，然后axis会被修改为[i,i+1,…,n-1,n]，例如reduceToEnd=True，axis=[2,4]，x的维度为7，则axis会被修改为[2,3,4,5,6]。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦