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 推理产品xAtlas 推理系列产品√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、BFLOAT16NDksize属性最大池化的窗口大小。INT64-strides属性窗口移动的步长。INT64-padding属性指定padding的模式。STRING-pads属性每一条边补充的层数。INT64-dilations属性控制窗口中元素的步幅。INT64-ceilMode属性计算输出形状的取整模式。INT64-data_fromat属性支持的数据格式STRING-y输出输出的张量。FLOAT16、FLOAT、BFLOAT16ND约束说明值域限制说明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创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考