AvgPool

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	×
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

• 算子功能：对输入Tensor进行窗口为$kH\ast kW$、步长为$sH\ast sW$的二维平均池化操作，其中$k$为kernelSize，表示池化窗口的大小，$s$为stride，表示池化操作的步长。

• 计算公式： 输入input（$N,C,H,W$）、输出out（$N,C,H_{out},W_{out}$）和池化步长（$strides$）、池化窗口大小($kH,kW$)的关系是

  $$H_{out}=\lfloor \frac{H_{in}+2\ast paddings[0]-kH}{strides[0]}+1\rfloor$$

  $$W_{out}=\lfloor \frac{W_{in}+2\ast paddings[1]-kW}{strides[1]}+1\rfloor$$

  $$out(N_i,C_i,h,w)=\frac{1}{kH\ast kW}\sum _{m=0}^{kH-1}\sum _{n=0}^{kW-1}input(N_i,C_i,strides[0]\ast h+m,strides[1]\ast w+n)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入tensor，公式中的input	FLOAT、FLOAT16、BFLOAT16	ND
y	输出	输出tensor，公式中的output	FLOAT、FLOAT16、BFLOAT16	ND
ksize	属性	表示池化窗口大小，公式中的`kernelSize`。	INT64	-
strides	属性	表示池化操作的步长，公式中的`stride`。	INT64	-
pads	属性	表示在输入的D、H方向上pads补0的层数，公式中的`padding`。	INT64	-
data_format	可选属性	表示输入的数据类型，默认为NHWC	STRING	-

约束说明

无。

调用说明

调用方式	样例代码	说明
图模式接口	test_geir_avg_pool	通过IR AvgPool构图方式调用AvgPool算子。