ResizeUpsampleTrilinear

产品支持情况

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

功能说明

• 算子功能：对由多个输入通道组成的输入信号应用三线性插值算法进行上采样。
• 计算公式：

  • 核心算法逻辑：

    1. 将目标图像的每一个点映射回原图，得到一个带小数点的坐标。
    2. 根据这个浮点数坐标，计算前后相邻的原始图像的点。
    3. 分别计算相邻点到对应目标点的权重，按照权重相乘累加即可得到目标点值。

  • 具体计算逻辑： 缩放方式分为角对齐和边对齐，角对齐（alignCorners为true）表示按照原始图片左上角像素中心点对齐，边对齐（alignCorners为false）表示按照原始图片左上角顶点及两条边对齐，在计算缩放系数和坐标位置时存在差异。对于一个三维插值点$(N,C,D,H,W)$，则有以下公式：

    $$scale\_d=\{\begin{array}{ll}(self.dim[2]-1)/(outputSize[0]-1)& alignCorners=true\\ 1/scales\_d& alignCorners=false\&scales\_d>0\\ self.dim[2]/outputSize[0]& alignCorners=false\end{array}$$

    $$scale\_h=\{\begin{array}{ll}(self.dim[3]-1)/(outputSize[1]-1)& alignCorners=true\\ 1/scales\_h& alignCorners=false\&scales\_h>0\\ self.dim[3]/outputSize[1]& alignCorners=false\end{array}$$

    $$scale\_w=\{\begin{array}{ll}(self.dim[4]-1)/(outputSize[2]-1)& alignCorners=true\\ 1/scales\_w& alignCorners=false\&scales\_w>0\\ self.dim[4]/outputSize[2]& alignCorners=false\end{array}$$

    因此，对于output的某个方向上的点p(x,y,z)，映射回原始图像中的点记为q(x',y',z')，则有关系：

    $$x^{\prime }=\{\begin{array}{ll}x\ast scale\_d& alignCorners=true\\ MAX(0,(x+0.5)\ast scale\_d-0.5)& alignCorners=false\end{array}$$

    $$y^{\prime }=\{\begin{array}{ll}y\ast scale\_h& alignCorners=true\\ MAX(0,(y+0.5)\ast scale\_h-0.5)& alignCorners=false\end{array}$$

    $$z^{\prime }=\{\begin{array}{ll}z\ast scale\_w& alignCorners=true\\ MAX(0,(z+0.5)\ast scale\_w-0.5)& alignCorners=false\end{array}$$

    • 记：

      $$x_0=int(x^{\prime }),x_1=int(x^{\prime })+1,lambd{a}_0=x_1-x^{\prime },lambd{a}_1=1-lambd{a}_0$$

      $$y_0=int(y^{\prime }),y_1=int(y^{\prime })+1,lambd{b}_0=y_1-y^{\prime },lambd{b}_1=1-lambd{b}_0$$

      $$z_0=int(z^{\prime }),z_1=int(z^{\prime })+1,lambd{c}_0=z_1-z^{\prime },lambd{c}_1=1-lambd{c}_0$$

    • 则有以下公式：

      $$V(p_{x,y,z})=V(p_{x0,y0,z0})\ast lambd{a}_0\ast lambd{b}_0\ast lambd{c}_0+V(p_{x0,y0,z1})\ast lambd{a}_0\ast lambd{b}_0\ast lambd{c}_1+V(p_{x0,y1,z0})\ast lambd{a}_0\ast lambd{b}_1\ast lambd{c}_0+V(p_{x0,y1,z1})\ast lambd{a}_0\ast lambd{b}_1\ast lambd{c}_1+V(p_{x1,y0,z0})\ast lambd{a}_1\ast lambd{b}_0\ast lambd{c}_0+V(p_{x1,y0,z1})\ast lambd{a}_1\ast lambd{b}_0\ast lambd{c}_1+V(p_{x1,y1,z0})\ast lambd{a}_1\ast lambd{b}_1\ast lambd{c}_0+V(p_{x1,y1,z1})\ast lambd{a}_1\ast lambd{b}_1\ast lambd{c}_1$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
input	输入	表示进行上采样的输入张量，对应公式中的`self`。	FLOAT32、FLOAT16、BFLOAT16	ND
output_size	可选属性	表示指定`output`在D、H和W维度上的空间大小，对应公式中的`outputSize`。size为3，且各元素均大于零。 默认值为空。	LISTINT	-
align_corners	可选属性	决定是否对齐角像素点，对应公式中的`alignCorners`。align_corners为true，则输入和输出张量的角像素点会被对齐，否则输入和输出张量的左上角顶点及两条边对齐。 默认值为false。	BOOL	-
scales_d	可选属性	指定空间大小的depth维度乘数，对应公式中的`scales_d`。 默认值为空。	FLOAT32	-
scales_h	可选属性	指定空间大小的height维度乘数，对应公式中的`scales_h`。 默认值为空。	FLOAT32	-
scales_w	可选属性	指定空间大小的width维度乘数，对应公式中的`scales_w`。 默认值为空。	FLOAT32	-
output	输出	表示采样后的输出张量，对应公式描述中的`output`。数据类型和数据格式与入参`input`的数据类型和数据格式保持一致。	FLOAT32、FLOAT16、BFLOAT16	ND

Atlas 推理系列产品：

• 输入和输出的数据类型不支持BFLOAT16。
• 入参input不支持输入inf、-inf。

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_upsample_trilinear3d	通过aclnnUpsampleTrilinear3d接口方式调用ResizeUpsampleTrilinear算子。