CANN ops-nn MaxPool3D算子

MaxPool3D

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn

产品支持情况

产品	是否支持
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) = \max\limits_{{k\in[0,k_{D}-1],m\in[0,k_{H}-1],n\in[0,k_{W}-1]}}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算子。

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn