AvgPoolFusion
概述
在输入tensor上应用 2D 平均池化。支持int8量化输入。
该算子对应的NodeType为NODE_TYPE_AVGPOOL_FUSION。
参数限制:当padMode==PAD_MODE_PAD时,padList的数值必须大于等于0。其他情况下padding的数值必须为0。
输入:
- x,n维张量。
输出:
- output, 输出平均池化后的张量。
起始版本: 3.2
相关模块:NNRt
汇总
Public 属性
名称 | 描述 |
---|---|
long[] kernelSize | 用来取平均值的kernel大小,是一个长度为2的int数组[kernel_height,kernel_weight], 第一个数表示kernel高度,第二个数表示kernel宽度。 |
long[] strides | kernel移动的距离,是一个长度为2的int数组[stride_height,stride_weight], 第一个数表示高度上的移动步幅,第二个数表示宽度上的移动步幅。 |
long[] pad | x周围的填充,是一个长度为4的int数组[top,bottom,left,right],并且以最近邻的值填充。 |
enum PadModepadMode | 填充模式 |
enum RoundModeroundMode | 取整数的算法 |
enum Formatformat | 运算时的数据排列排列,详情请参考:Format |
boolean global | 是否是全局池化 |
enum ActivationTypeactivationType | 激活函数,详情请参考:ActivationType |
类成员变量说明
activationType
enum ActivationType AvgPoolFusion::activationType
描述 激活函数,详情请参考:ActivationType
format
enum Format AvgPoolFusion::format
描述 运算时的数据排列排列,详情请参考:Format
global
boolean AvgPoolFusion::global
描述 是否是全局池化
kernelSize
long [] AvgPoolFusion::kernelSize
描述 用来取平均值的kernel大小,是一个长度为2的int数组[kernel_height,kernel_weight], 第一个数表示kernel高度,第二个数表示kernel宽度。
pad
long [] AvgPoolFusion::pad
描述 x周围的填充,是一个长度为4的int数组[top,bottom,left,right],并且以最近邻的值填充。
padMode
enum PadMode AvgPoolFusion::padMode
描述 填充模式
roundMode
enum RoundMode AvgPoolFusion::roundMode
描述 取整数的算法
strides
long [] AvgPoolFusion::strides
描述 kernel移动的距离,是一个长度为2的int数组[stride_height,stride_weight], 第一个数表示高度上的移动步幅,第二个数表示宽度上的移动步幅。