AvgPoolFusion
概述
在输入tensor上应用 2D 平均池化。支持int8量化输入。
该算子对应的NodeType为NODE_TYPE_AVGPOOL_FUSION。
参数限制:当padMode==PAD_MODE_PAD时,padList的数值必须大于等于0。其他情况下padding的数值必须为0。
输入:
- x,n维张量。
输出:
- output, 输出平均池化后的张量。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
名称 | 描述 |
---|---|
kernelSize | long[] 用来取平均值的kernel大小 |
strides | long[] kernel移动的距离 |
pad | long[] x周围的填充 |
padMode | enum PadMode 填充模式 |
roundMode | enum RoundMode 取整数的算法 |
format | enum Format 运算时的数据排列排列,详情请参考:Format |
global | boolean 是否是全局池化 |
activationType | enum ActivationType 激活函数,详情请参考: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], 第一个数表示高度上的移动步幅,第二个数表示宽度上的移动步幅。