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]