AdaptiveMaxPool2d

产品支持情况

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

功能说明

• 算子功能：根据输入的output_size计算每次kernel的大小，对输入x进行3维最大池化操作，输出池化后的值y和索引indices。

• 计算公式：

  y tensor中对于DHW轴上每个位置为$(l,m,n)$的元素来说，其计算公式为：

  $$H_{left}^m=floor((m\ast H)/H_o)$$

  $$H_{right}^m=ceil(((m+1)\ast H)/H_o)$$

  $$W_{left}^n=floor((n\ast W)/W_o)$$

  $$W_{right}^n=ceil(((n+1)\ast W)/W_o)$$

  $$y(N,C,m,n)=\underset{j\in [H_{left}^m,H_{right}^m],k\in [W_{left}^n,W_{right}^n]}{\max }input(N,C,j,k)$$

  $$indices(N,C,m,n)=\underset{j\in [H_{left}^m,H_{right}^m],k\in [W_{left}^n,W_{right}^n]}{\mathrm{argmax}}input(N,C,j,k)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行AdaptiveMaxPool2d计算的入参。H轴W轴2个维度的乘积不能大于int32的最大表示。数据类型与出参y的保持一致。 shape为[N, C, Hin, Win]。	FLOAT、FLOAT16、BFLOAT16	ND
output_size	属性	表示输出结果在H，W维度上的空间大小。数据类型与入参x的保持一致。 值为[Hout, Wout]。	INT64	-
y	输出	待进行AdaptiveMaxPool2d计算的出参。shape与出参indices的保持一致，数据类型与入参x的保持一致。 shape为[N, C, Hout, Wout]。	FLOAT、FLOAT16、BFLOAT16	ND
indices	输出	表示y元素在输入x中的索引位置。shape与出参y的保持一致。 shape为[N, C, Hout, Wout]。	INT32	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_adaptive_max_pool2d.cpp	通过aclnnAdaptiveMaxPool2d接口方式调用AdaptiveMaxPool2d算子。
图模式	test_geir_adaptive_max_pool2d.cpp	通过GE IR构图方式调用AdaptiveMaxPool2d算子。