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], 第一个数表示高度上的移动步幅,第二个数表示宽度上的移动步幅。