MaxPoolV3

产品支持情况

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

功能说明

• 接口功能： 对于dim=3或4维的输入张量，进行最大池化（max pooling）操作。
• 计算公式：

  • 当ceilMode=False时，out tensor的shape中H和W维度推导公式：

    $$[H_{out},W_{out}]=[\lfloor \frac{H_{in}+padding\_siz{e}_{Htop}+padding\_siz{e}_{Hbottom}-dilation\_size\times (k_h-1)-1}{s_h}\rfloor +1,\lfloor \frac{W_{in}+padding\_siz{e}_{Wleft}+padding\_siz{e}_{Wright}-dilation\_size\times (k_w-1)-1}{s_w}\rfloor +1]$$

  • 当ceilMode=True时，out tensor的shape中H和W维度推导公式：

    $$[H_{out},W_{out}]=[\lceil \frac{H_{in}+padding\_siz{e}_{Htop}+padding\_siz{e}_{Hbottom}-dilation\_size\times (k_h-1)-1}{s_h}\rceil +1,\lceil \frac{W_{in}+padding\_siz{e}_{Wleft}+padding\_siz{e}_{Wright}-dilation\_size\times (k_w-1)-1}{s_w}\rceil +1]$$

    • 滑窗左上角起始位处在下或右侧pad填充位上或者界外（无法取到有效值）时，舍弃该滑窗结果，在上述推导公式基础上对应空间轴shape需减去1：

      $$\{\begin{array}{ll}{H}_{out}=H_{out}-1& \text{if }(H_{out}-1)\ast s_h>=H_{in}+padding\_siz{e}_{Htop}\\ W_{out}=W_{out}-1& \text{if }(W_{out}-1)\ast s_w>=W_{in}+padding\_siz{e}_{Wleft}\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
kernelShape	输入	最大池化的窗口大小。	INT64	ND
strides	输入	窗口移动的步长。	INT64	ND
autoPad	输入	指定padding的方式。	INT64	ND
pads	输入	沿着空间轴方向开始和结束的位置填充，对应公式中的padding_size。	INT64	ND
dilations	输入	沿着核空间轴方向的膨胀值，对应公式中的dilation_size。	INT64	-
ceilMode	输入	计算输出形状的取整模式。	INT64	-
out	输出	输出的tensor。	FLOAT16、FLOAT、BF16、INT32、INT64、UINT8、INT16、INT8、UINT16。	-

约束说明

• 值域限制说明：
  • kernelShape：对应公式中的k_h和k_w，长度为1或2，且数组元素必须都大于0。
  • strides：对应公式中的s_h和s_w，数组长度为0、1或2，且数组元素必须都大于0。当数组长度为0时，strides取默认值为1。
  • autoPad：其中0代表"NOTSET"，并且只支持数值0。
  • pads：长度为0、1、2或4。当数组长度为0时，不进行填充。当数组长度为1时，H_top、H_bottom、W_left、W_right填充同一个值。当数组长度为2时，H_top、H_bottom分别填充数组第1个值，W_left、W_right分别填充数组第2个值。当数组长度为4时，按[H_top、W_left、H_bottom、W_right]位置关系进行填充。单个空间轴方向填充量之和需小于等于对应方向kernelShape。
  • dilations：只支持数值为1的输入场景。长度为0、1、2或4。
  • ceilMode：取值为0时，代表False，向下取整；非0值时，代表True，向上取整。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_max_pool	通过aclnnMaxPool接口方式调用MaxPoolV3算子。