aclnnAdaptiveMaxPool3d

产品支持情况

产品	是否支持
[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]	√
[object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]	√
[object Object]Atlas 200I/500 A2 推理产品[object Object]	×
[object Object]Atlas 推理系列产品 [object Object]	×
[object Object]Atlas 训练系列产品[object Object]	×

功能说明

• 算子功能：根据输入的outputSize计算每次kernel的大小，对输入self进行3维最大池化操作，输出池化后的值outputOut和索引indicesOut。aclnnAdaptiveMaxPool3d与aclnnMaxPool3d的区别在于，只需指定outputSize大小，并按outputSize的大小来划分pooling区域。

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

  $$D^{l}_{left} = floor((l*D)/D_o)$$ $$D^{l}_{right} = ceil(((l+1)*D)/D_o)$$ $$H^{m}_{left} = floor((m*H)/H_o)$$ $$H^{m}_{right} = ceil(((m+1)*H)/H_o)$$ $$W^{n}_{left} = floor((n*W)/W_o)$$ $$W^{n}_{right} = ceil(((n+1)*W)/W_o)$$ $$outputOut(N,C,l,m,n)=\underset {i \in [D^{l}_{left}, D^{l}_{right}],j\in [H^m_{left},H^m_{right}], k \in [W^n_{left},W^n_{right}] }{max} input(N,C,i,j,k)$$ $$indicesOut(N,C,l,m,n)=\underset {i \in [D^{l}_{left}, D^{l}_{right}],j\in [H^m_{left},H^m_{right}], k \in [W^n_{left},W^n_{right}] }{argmax} input(N,C,i,j,k)$$

函数原型

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

• aclnnStatus aclnnAdaptiveMaxPool3dGetWorkspaceSize(const aclTensor* self, const aclIntArray* outputSize, aclTensor* outputOut, aclTensor* indicesOut, uint64_t* workspaceSize, aclOpExecutor** executor)
• aclnnStatus aclnnAdaptiveMaxPool3d(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream)

aclnnAdaptiveMaxPool3dGetWorkspaceSize

• 参数说明：

  • self（aclTensor*，计算输入）：输入Tensor，Device侧的aclTensor。支持undefined，undefined支持NCDHW，当输入是4维时，在0维度处补1，内部按照NCDHW处理。D轴H轴W轴3个维度的乘积$D*H*W$不能大于int32的最大表示。数据类型支持BFLOAT16、FLOAT16、FLOAT32，且数据类型与outputOut的数据类型保持一致。
  • outputSize（aclIntArray*，计算输入）：Host侧的aclIntArray，size大小为3。表示输出结果在$D_o$，$H_o$，$W_o$维度上的空间大小。数据类型支持INT32和INT64。outputSize中元素值不可小于0。
  • outputOut（aclTensor*，计算输出）：输出Tensor，是Device侧的aclTensor。池化后的结果。shape与indicesOut保持一致。undefined支持NCDHW。数据类型支持BFLOAT16、FLOAT16、FLOAT32，且数据类型与self的数据类型一致。
  • indicesOut（aclTensor*，计算输出）：输出Tensor，是Device侧的aclTensor。indicesOut表示outputOut元素在输入self中的索引位置。shape与outputOut保持一致，undefined支持NCDHW。数据类型支持INT32。
  • workspaceSize（uint64_t*，出参）：返回用户需要在npu Device侧申请的workspace大小。
  • executor（aclOpExecutor**，出参）：返回op执行器，包含了算子计算流程。

• 返回值：

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

  [object Object]

aclnnAdaptiveMaxPool3d

• 参数说明：

  • workspace（void*，入参）：在npu device侧申请的workspace内存地址。
  • workspaceSize（uint64_t，入参）：在npu device侧申请的workspace大小，由第一段接口aclnnAdaptiveMaxPool3dGetWorkspaceSize获取。
  • executor（aclOpExecutor*，入参）：op执行器，包含了算子计算流程。
  • stream（aclrtStream，入参）：指定执行任务的Stream。

• 返回值：

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

约束说明

Shape描述：

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

调用示例

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