aclnnUpsampleNearestExact3d

产品支持情况

[object Object]undefined

功能说明

• 接口功能：对由多个输入通道组成的输入信号应用最近邻插值算法进行上采样。如果输入shape为（N，C，D，H，W），则输出shape为（N，C，outputSize[0]，outputSize[1]，outputSize[2]）。
• 计算公式：$$d_{src} = min(floor((d_{dst} + 0.5) / scalesD), D - 1),scalesD = outputSize[0] / D$$ $$h_{src} = min(floor((h_{dst} + 0.5) / scalesH), H - 1),scalesH = outputSize[1] / H$$ $$w_{src} = min(floor((w_{dst} + 0.5) / scalesW), W - 1),scalesW = outputSize[2] / W$$ $$out(N, C, d_{dst},h_{dst}, w_{dst}) = self(N, C, d_{src},h_{src}, w_{src})$$

函数原型

每个算子分为，必须先调用“aclnnUpsampleNearestExact3dGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小，再调用“aclnnUpsampleNearestExact3d”接口执行计算。

aclnnUpsampleNearestExact3dGetWorkspaceSize

• 参数说明：

  [object Object]

  • [object Object]Atlas 推理系列产品[object Object]：

    参数[object Object]、[object Object]的数据类型仅支持FLOAT32、FLOAT16。

• 返回值：

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

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

  [object Object]

aclnnUpsampleNearestExact3d

• 参数说明：

  [object Object]

• 返回值：

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

约束说明

• 参数outputSize与参数scalesD、scalesH、scalesW，在使用时二选一，即：
  • 当入参scalesD、scalesH、scalesW，其中一个参数的值小于等于0时，使用入参outputSize的参数值。
  • 当入参scalesD、scalesH、scalesW的值都大于0时，使用入参scalesD、scalesH、scalesW的参数值，且$outputSize=[floor(self\_D * scalesD)，floor(self\_H * scalesH)，floor(self\_W * scalesW)]$。
• 确定性计算：
  • aclnnUpsampleNearestExact3d默认确定性实现。

调用示例

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