UpsampleBicubic2dAA

产品支持情况

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

功能说明

• 算子功能：对由多个输入通道组成的输入信号应用双三次抗锯齿算法进行上采样。如果输入Tensor x的shape为(N, C, H, W)，则输出Tensor out的shape为(N, C, outputSize[0], outputSize[1])。

• 计算公式：对于一个二维插值点$(N,C,h,w)$，插值$out(N,C,h,w)$可以表示为：

  $$out(N,C,h,w)=\sum _{i=0}^{kW}\sum _{j=0}^{kH}W(i,j)\ast f(h_i,w_j)$$

  $$scaleH=\{\begin{array}{ll}(x.dim(2)-1)/(outputSize[0]-1)& alignCorners=true\\ 1/scalesH& alignCorners=false\&scalesH>0\\ x.dim(2)/outputSize[0]& otherwise\end{array}$$

  $$scaleW=\{\begin{array}{ll}(x.dim(3)-1)/(outputSize[1]-1)& alignCorners=true\\ 1/scalesW& alignCorners=false\&scalesW>0\\ x.dim(3)/outputSize[1]& otherwise\end{array}$$

  其中：

  • alignCorners为true，表示输入和输出张量的角像素点对齐；alignCorners为false，表示输入和输出张量的边像素点对齐。

  • i和j是$W(i,j)$的索引变量。

  • 如果$scaleH>=1$，则$kH=1/scaleH$，否则$kH=4$

  • 如果$scaleW>=1$，则$kW=1/scaleW$，否则$kW=4$

  • $h_i=|h|+i$

  • $w_j=|w|+j$

  • $f(h_i,w_j)$是原图像在$(h_i,w_j)$的像素值

  • $W(i,j)$是双三次抗锯齿插值的权重，定义为：

    $$W(d)=\{\begin{array}{ll}(a+2)|d{|}^3-(a+3)|d{|}^2+1& |d|\le 1\\ a|d{|}^3-5a|d{|}^2+8a|d|-4a& 1<|d|<2\\ 0& otherwise\end{array}$$

    其中：

    • 抗锯齿场景$a=-0.5$。
    • $d=|(h,w)-(h_i,w_j)|$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	表示进行上采样的输入张量，对应公式中的`x`。数据类型与出参`y`的数据类型一致。	FLOAT32、FLOAT16、BFLOAT16	ND
output_size	属性	表示指定`y`在H和W维度上的空间大小，对应公式中的`outputSize`。size为2。	INT64	-
align_corners	可选属性	决定是否对齐角像素点，对应公式中的`alignCorners`。align_corners为true，则输入和输出张量的角像素点会被对齐，否则不对齐。 默认值为false。	BOOL	-
scales_h	可选属性	指定空间大小的height维度乘数，对应公式中的`scalesH`。 默认值为0.0。	FLOAT32	-
scales_w	可选属性	指定空间大小的width维度乘数，对应公式中的`scalesW`。 默认值为0.0。	FLOAT32	-
y	输出	表示采样后的输出张量，对应公式中的`out`。数据类型与入参`x`的数据类型一致。	FLOAT32、FLOAT16、BFLOAT16	ND

• Kirin X90 处理器系列产品、Kirin 9030 处理器系列产品：不支持BFLOAT16。

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_upsample_bicubic2d_aa	通过aclnnUpsampleBicubic2dAA接口方式调用UpsampleBicubic2dAA算子。