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}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	4维输入tensor。	FLOAT16、BF16、FLOAT32、DOUBLE、INT32、INT64、UINT8、INT16、INT8、UINT16、QINT8	NHWC、NCHW
ksize	属性	最大池化的窗口大小，长度为4的列表。N和C维度的ksize必须为1，H和W维度的ksize必须大于0。	ListInt	-
strides	属性	滑动窗口的步长，长度为4的列表。N和C维度的stride必须为1。	ListInt	-
padding_mode	属性	指定padding的方式。支持"SAME"、"VALID"、"CALCULATED"，默认为"CALCULATED"。"SAME"时补零使输出shape等于ceil(input shape / stride)；"VALID"时不补零；"CALCULATED"时使用pads计算输出shape。	String	-
pads	属性	沿着空间轴方向开始和结束的位置填充，对应公式中的padding_size。长度为4，指定[top, bottom, left, right]的填充量。仅在padding_mode为"CALCULATED"时生效。默认值为{0,0,0,0}。	ListInt	-
data_format	属性	输入输出的数据格式，支持"NHWC"和"NCHW"，默认为"NCHW"。	String	-
global_pooling	属性	是否使用全局池化。为True时忽略ksize和pads，输出H和W均为1。默认False。	Bool	-
ceil_mode	属性	计算输出形状的取整模式。True时向上取整，False时向下取整。padding_mode为"SAME"或"VALID"时仅支持False。默认False。	Bool	-
y	输出	4维输出tensor，与输入x具有相同的数据类型和格式。	FLOAT16、BF16、FLOAT32、DOUBLE、INT32、INT64、UINT8、INT16、INT8、UINT16、QINT8	NHWC、NCHW

约束说明

• 值域限制说明：
  • ksize：长度为4的列表。H和W维度的ksize必须大于0，N和C维度的ksize必须为1。H和W维度的ksize乘积应小于等于255。
  • strides：长度为4的列表。N和C维度的stride必须为1，H和W维度的stride必须大于0。
  • padding_mode：支持"SAME"、"VALID"、"CALCULATED"三种模式。
  • pads：长度为4，按[top, bottom, left, right]指定填充量，仅在padding_mode为"CALCULATED"时生效。pads应大于等于0且小于对应的kernel size。
  • global_pooling：为True时，输出H和W均为1，ksize和pads被忽略。
  • ceil_mode：padding_mode为"SAME"或"VALID"时仅支持False。

调用说明

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