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。
起始版本: 3.2
相关模块:NNRt
汇总
Public 属性
名称 | 描述 |
---|---|
boolean keepDims | 维度是否保持不变。 |
enum ReduceModemode | 减小张量维度的算法,详情请参考:ReduceMode。 |
boolean reduceToEnd | 如果为true,则从axis取第一个元素并设置为i, 然后axis会被修改为[i,i+1,...,n-1,n], 例如reduceToEnd=True,axis=[2,4],x的维度为7,则axis会被修改为[2,3,4,5,6] |
float coeff | 系数 |
类成员变量说明
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]