MaxPool3D

产品支持情况

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

功能说明

• 接口功能： 对于输入信号的输入通道，提供3维最大池化（Max pooling）操作。
• 计算公式：

  • output tensor中每个元素的计算公式：

    $$out(N_i,C_j,d,h,w)=\underset{k\in [0,k_D-1],m\in [0,k_H-1],n\in [0,k_W-1]}{\max }input(N_i,C_j,stride[0]\times d+k,stride[1]\times h+m,stride[2]\times w+n)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入的张量。	FLOAT16、FLOAT、BFLOAT16	ND
ksize	属性	最大池化的窗口大小。	INT64	-
strides	属性	窗口移动的步长。	INT64	-
padding	属性	指定padding的模式。	STRING	-
pads	属性	每一条边补充的层数。	INT64	-
dilations	属性	控制窗口中元素的步幅。	INT64	-
ceilMode	属性	计算输出形状的取整模式。	INT64	-
data_fromat	属性	支持的数据格式	STRING	-
y	输出	输出的张量。	FLOAT16、FLOAT、BFLOAT16	ND

约束说明

• 值域限制说明：
  • ksize：数组长度必须为5，且N和C维度对应的值必须为1。
  • strides：数组长度必须为5，且N和C维度对应的值必须为1。
  • padding：只支持三种模式：“SAME”、“VALID”、“CALCULATED”。
  • pads：该参数仅在padding模式为“CALCULATED”时生效。
  • dilations：数组长度必须为5，且N和C维度对应的值必须为1。
  • ceilMode：取值为0时，代表False，向下取整；非0值时，代表True，向上取整，该参数仅在padding模式为“CALCULATED”时生效。

调用说明

调用方式	样例代码	说明
图模式接口	test_max_pool_3d	通过IRMaxPool3D构图方式调用MaxPool3D算子。