AdaptiveMaxPool3d

产品支持情况

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

功能说明

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

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

  $$D_{left}^l=floor((l\ast D)/D_o)$$

  $$D_{right}^l=ceil(((l+1)\ast D)/D_o)$$

  $$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,l,m,n)=\underset{i\in [D_{left}^l,D_{right}^l],j\in [H_{left}^m,H_{right}^m],k\in [W_{left}^n,W_{right}^n]}{\max }input(N,C,i,j,k)$$

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

参数说明

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

• Kirin X90/Kirin 9030 处理器系列产品: 不支持BFLOAT16。

约束说明

Shape描述：

• self.shape = (N, C, Din, Hin, Win)
• outputSize = [Dout, Hout, Wout]
• outputOut.shape = (N, C, Dout, Hout, Wout)
• indicesOut.shape = (N, C, Dout, Hout, Wout)

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_adaptive_max_pool3d.cpp	通过aclnnAdaptiveMaxPool3d接口方式调用AdaptiveMaxPool3d算子。