ResizeNearestNeighborV2

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：对由多个输入通道组成的输入信号应用最近邻插值算法进行上采样。如果输入shape为(N, C, H, W)，则输出shape为(N, C, size[0], size[1])。

• 计算公式：

  $$y(N,C,H,W)=x(N,C,min(floor(H\ast scaleH),H-1),min(floor(W\ast scaleW),W-1))$$

  $$scaleH=x\_H/size[0]$$

  $$scaleW=x\_W/size[1]$$

参数说明

参数名	输入/输出	描述	数据类型	数据格式
x	输入	输入图像的四维Tensor，对应公式中`x`。	FLOAT16、FLOAT32、BFLOAT16	NCHW、NHWC
size	输入	输出图像的高和宽。	INT32	-
y	输出	最近邻插值后的图像，对应公式中的`y`。	FLOAT16、FLOAT32、BFLOAT16	NCHW、NHWC

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn	test_aclnn_upsample_nearest2d.cpp	通过aclnnUpsampleNearest2d接口方式调用ResizeNearestNeighborV2算子。
图模式	-	通过算子IR构图方式调用ResizeNearestNeighborV2算子。