昇腾社区首页
中文
注册
开发者
下载

aclnnUpsampleNearestExact2d

产品支持情况

[object Object]undefined

功能说明

  • 接口功能:对由多个输入通道组成的输入信号应用最近邻插值算法进行上采样。如果输入shape为(N,C,H,W),则输出shape为(N,C,outputSize[0],outputSize[1])。
  • 计算公式:hsrc=min(floor((hdst+0.5)/scalesH),H1),scalesH=outputSize[0]/Hh_{src} = min(floor((h_{dst} + 0.5) / scalesH), H - 1),scalesH = outputSize[0] / H wsrc=min(floor((wdst+0.5)/scalesW),W1),scalesW=outputSize[1]/Ww_{src} = min(floor((w_{dst} + 0.5) / scalesW), W - 1),scalesW = outputSize[1] / W out(N,C,hdst,wdst)=self(N,C,hsrc,wsrc)out(N, C, h_{dst}, w_{dst}) = self(N, C, h_{src}, w_{src})

函数原型

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

[object Object]
[object Object]

aclnnUpsampleNearestExact2dGetWorkspaceSize

  • 参数说明:

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

      参数self、out的数据类型仅支持FLOAT32、FLOAT16。

  • 返回值:

    aclnnStatus:返回状态码,具体参见

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

    [object Object]

aclnnUpsampleNearestExact2d

  • 参数说明:

    [object Object]
  • 返回值:

    aclnnStatus:返回状态码,具体参见

约束说明

  • 参数outputSize的H轴和W轴与参数scalesH和参数scalesW,在使用时二选一,即:
    • 当入参scalesH或入参scalesW的值小于等于0时,使用入参outputSize中对应轴的参数值。
    • 当入参scalesH或入参scalesW的值大于0时,使用入参scalesH或入参scalesW的参数值,即outputSize对应轴的值为floor(self_HscalesH)floor(self\_H * scalesH),或者floor(self_WscalesW)floor(self\_W * scalesW)
  • 确定性计算:
    • aclnnUpsampleNearestExact2d默认确定性实现。

调用示例

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

[object Object]