aclnnMaxPool3dWithArgmax

产品支持情况

[object Object]undefined

功能说明

• 算子功能：

  • 对于输入信号的输入通道，提供3维最大池化（Max pooling）操作，输出池化后的值out和索引indices。
  • 输入dims的描述：N - 批次，C - 通道，D - 深度，W - 宽度，H - 高度。
  • 当D * H * W超过int32时，建议在模型尺寸上分割D轴。

• 计算公式：

  • 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)$$

  • out tensor的shape推导公式（默认ceilMode=false，即向下取整）：

$$[N, C, D_{out}, H_{out}, W_{out}]=[N,C,\lfloor{\frac{D_{in}+2 \times {padding[0] - dilation[0] \times(kernelSize[0] - 1) - 1}}{stride[0]}}\rfloor + 1,\\ \lfloor{\frac{H_{in}+2 \times {padding[1] - dilation[1] \times(kernelSize[1] - 1) - 1}}{stride[1]}}\rfloor + 1, \\ \lfloor{\frac{W_{in}+2 \times {padding[2] - dilation[2] \times(kernelSize[2] - 1) - 1}}{stride[2]}}\rfloor + 1]$$

• out tensor的shape推导公式（默认ceilMode=true，即向上取整）：

$$[N, C, D_{out}, H_{out}, W_{out}]=[N,C,\lceil{\frac{D_{in}+2 \times {padding[0] - dilation[0] \times(kernelSize[0] - 1) - 1}}{stride[0]}}\rceil + 1,\\ \lceil{\frac{H_{in}+2 \times {padding[1] - dilation[1] \times(kernelSize[1] - 1) - 1}}{stride[1]}}\rceil + 1, \\ \lceil{\frac{W_{in}+2 \times {padding[2] - dilation[2] \times(kernelSize[2] - 1) - 1}}{stride[2]}}\rceil + 1]$$

函数原型

每个算子分为，必须先调用“aclnnMaxPool3dWithArgmaxGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnMaxPool3dWithArgmax”接口执行计算。

aclnnMaxPool3dWithArgmaxGetWorkspaceSize

• 参数说明：

  [object Object]
  • [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]、[object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]：[object Object] 元素值仅支持为1；[object Object] 数据类型不支持INT64。输入数据排布不支持NDHWC。depth * height * width 不支持大于 max int32。

• 返回值：

  aclnnStatus: 返回状态码，具体参见。

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

aclnnMaxPool3dWithArgmax

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

约束说明

• 确定性计算：aclnnMaxPool3dWithArgmax默认确定性实现。

• kernelSize、stride、padding、dilation、ceilMode参数需要保证输出out shape中不存在小于1的轴。

• 当ceilMode为True的时候，如果滑动窗口全部在右侧padding区域上，这个输出结果将被忽略。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。