UpsampleBilinear2dGrad
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | × |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | × |
功能说明
- 算子功能:对由多个输入通道组成的输入信号应用2D双线性上采样的反向传播。
- 计算公式:
-
正向的核心算法逻辑:
- 将目标图像的每一个点映射回原图,得到一个带小数点的坐标。
- 根据这个浮点数坐标,计算前后相邻的原始图像的点。
- 分别计算相邻点到对应目标点的权重,按照权重相乘累加即可得到目标点值。
-
具体计算逻辑: 缩放方式分为角对齐和边对齐,角对齐(alignCorners为true)表示按照原始图片左上角像素中心点对齐,边对齐(alignCorners为false)表示按照原始图片左上角顶点及两条边对齐,在计算缩放系数和坐标位置时存在差异。对于一个二维插值点(N,C,H,W)(N, C, H, W),则有以下公式:
scale_h={(inputSize[2]−1)/(outputSize[0]−1)alignCorners=true1/scales_halignCorners=false&scales_h>0inputSize[2]/outputSize[0]alignCorners=falsescale\_h =\begin{cases} (inputSize[2]-1) / (outputSize[0]-1) & alignCorners=true \\ 1 / scales\_h & alignCorners=false\&scales\_h>0\\ inputSize[2] / outputSize[0] & alignCorners=false \end{cases}
scale_w={(inputSize[3]−1)/(outputSize[1]−1)alignCorners=true1/scales_walignCorners=false&scales_w>0inputSize[3]/outputSize[1]alignCorners=falsescale\_w =\begin{cases} (inputSize[3]-1) / (outputSize[1]-1) & alignCorners=true \\ 1 / scales\_w & alignCorners=false\&scales\_w>0\\ inputSize[3] / outputSize[1] & alignCorners=false \end{cases}
因此,对于output的某个方向上的点p(x,y),映射回原始图像中的点记为q(x',y'),则有关系:
x′={x∗scale_halignCorners=trueMAX(0,(x+0.5)∗scale_h−0.5)alignCorners=falsex' =\begin{cases} x * scale\_h & alignCorners=true \\ MAX(0,{(x+0.5)*scale\_h-0.5}) & alignCorners=false \end{cases}
y′={y∗scale_walignCorners=trueMAX(0,(y+0.5)∗scale_w−0.5)alignCorners=falsey' =\begin{cases} y * scale\_w & alignCorners=true \\ MAX(0,{(y+0.5)*scale\_w-0.5}) & alignCorners=false \end{cases}
-
记:
x0=int(x′),x1=int(x′)+1,lambda0=x1−x′,lambda1=1−lambda0x_{0} =int(x'),x_{1} =int(x')+1, lambda_{0} = x_{1}-x', lambda_{1} = 1-lambda_{0}
y0=int(y′),y1=int(y′)+1,lambdb0=y1−y′,lambdb1=1−lambdb0y_{0} =int(y'),y_{1} =int(y')+1, lambdb_{0} = y_{1}-y', lambdb_{1} = 1-lambdb_{0}
-
则有以下公式:
V(px,y)=V(px0,y0)∗lambda0∗lambdb0+V(px0,y1)∗lambda0∗lambdb1+V(px1,y0)∗lambda1∗lambdb0+V(px1,y1)∗lambda1∗lambdb1{V(p_{x, y})} = {V(p_{x0, y0})} * {lambda_{0}} * {lambdb_{0}} + {V(p_{x0, y1})} * {lambda_{0}} * {lambdb_{1}} + {V(p_{x1, y0})} * {lambda_{1}} * {lambdb_{0}} + {V(p_{x1, y1})} * {lambda_{1}} * {lambdb_{1}}
-
假设:正向插值的输出图像out (x,y)(x, y)受原图像input (xi,yj)(x_i, y_j)影响,则有:
gradInput(xi,yj)+=gradOutput(x,y)∗lambda(xi,yj)∗lambdb(xi,yj)gradInput(x_i,y_j) += gradOutput(x,y) * lambda(x_i,y_j)* lambdb(x_i,y_j)
-
-
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| grad_output | 输入 | 表示反向计算的梯度Tensor,对应公式中的`gradOutput`。 | FLOAT32、FLOAT16、BFLOAT16 | ND |
| output_size | 属性 | 指定输入空间大小,对应公式中的`outputSize`。size为2,且各元素均大于0。表示指定`grad_output`在H和W维度上的空间大小。 | LISTINT | - |
| input_size | 属性 | 指定输出空间大小,对应公式中的`inputSize`。size为4,且各元素均大于零。表示输出`grad_input`分别在N、C、H和W维度上的空间大小。 | LISTINT | - |
| align_corners | 可选属性 |
|
BOOL | - |
| scales_h | 可选属性 |
|
FLOAT32 | - |
| scales_w | 可选属性 |
|
FLOAT32 | - |
| grad_input | 输出 | 表示反向计算的输出张量,对应公式中的`gradInput`。数据类型和数据格式与入参`grad_output`的数据类型和数据格式保持一致。 | FLOAT32、FLOAT16、BFLOAT16 | ND |
Atlas 推理系列产品、Atlas 200I/500 A2 推理产品:输入和输出的数据类型不支持BFLOAT16。
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_aclnn_upsample_bilinear2d_grad | 通过aclnnUpsampleBilinear2dBackwardV2接口方式调用UpsampleBilinear2dGrad算子。 |
| aclnn接口 | test_aclnn_upsample_linear1d_backward | 通过aclnnUpsampleLinear1dBackward接口方式调用UpsampleBilinear2dGrad算子。 |