MaxPoolExt2

产品支持情况

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

功能说明

• 接口功能： 对于4维的输入张量，进行最大池化（max pooling）操作。支持NHWC和NCHW两种数据格式，支持SAME和VALID两种填充模式。
• 计算公式：

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

    $$[H_{out},W_{out}]=[\lceil \frac{H_{in}}{s_h}\rceil ,\lceil \frac{W_{in}}{s_w}\rceil ]$$

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

    $$[H_{out},W_{out}]=[\lceil \frac{H_{in}-k_h+1}{s_h}\rceil ,\lceil \frac{W_{in}-k_w+1}{s_w}\rceil ]$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入tensor，维度为4，公式中的x。	FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8、UINT16	ND
ksize	属性	最大池化的窗口大小，长度为4的列表，对应公式中的k_h和k_w。	ListInt	-
strides	属性	窗口移动的步长，长度为4的列表，对应公式中的s_h和s_w。	ListInt	-
padding	属性	填充模式，取值为"SAME"或"VALID"。SAME模式表示输出大小为输入大小除以步长向上取整，VALID模式表示输出大小为输入大小减去窗口大小加1再除以步长向上取整。	String	-
data_format	属性	数据格式，支持"NHWC"或"NCHW"，默认为"NHWC"。	String	-
y	输出	输出tensor，与x同dtype，维度为4。	FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8、UINT16	ND

约束说明

• 值域限制说明：
  • x：输入tensor维度必须为4。
  • ksize：长度为4的列表，其中ksize[0]=1且ksize[3]=1（NHWC格式）或ksize[0]=1且ksize[1]=1（NCHW格式），ksize[1]和ksize[2]（或对应空间轴）必须大于0。
  • strides：长度为4的列表，其中strides[0]=1且strides[3]=1（NHWC格式）或strides[0]=1且strides[1]=1（NCHW格式），strides[1]和strides[2]（或对应空间轴）必须大于0。
  • padding：取值必须为"SAME"或"VALID"。
  • data_format：取值必须为"NHWC"或"NCHW"，默认为"NHWC"。
  • 输出y的dtype与输入x的dtype相同。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_max_pool_ext2	参见算子调用完成算子编译和验证。