开发者
资源
[object Object]

[object Object][object Object]undefined
[object Object]

正向最大池化的反向传播。

  • 输入tensor的推导公式:
    • 当ceilMode=False时,indices tensor的shape中H和W维度推导公式:

      [Hout,Wout]=[Hin+padding_sizeHtop+padding_sizeHbottomdilation_size×(kh1)1sh+1,Win+padding_sizeWleft+padding_sizeWrightdilation_size×(kw1)1sw+1][H_{out}, W_{out}]=[\lfloor{\frac{H_{in}+ padding\_size_{Htop} + padding\_size_{Hbottom} - {dilation\_size \times(k_h - 1) - 1}}{s_h}}\rfloor + 1,\lfloor{\frac{W_{in}+ padding\_size_{Wleft} + padding\_size_{Wright} - {dilation\_size \times(k_w - 1) - 1}}{s_w}}\rfloor + 1]
    • 当ceilMode=True时,out tensor的shape中H和W维度推导公式:

      [Hout,Wout]=[Hin+padding_sizeHtop+padding_sizeHbottomdilation_size×(kh1)1sh+1,Win+padding_sizeWleft+padding_sizeWrightdilation_size×(kw1)1sw+1][H_{out}, W_{out}]=[\lceil{\frac{H_{in}+ padding\_size_{Htop} + padding\_size_{Hbottom} - {dilation\_size \times(k_h - 1) - 1}}{s_h}}\rceil + 1,\lceil{\frac{W_{in}+ padding\_size_{Wleft} + padding\_size_{Wright} - {dilation\_size \times(k_w - 1) - 1}}{s_w}}\rceil + 1]
    • 滑窗左上角起始位处在下或右侧pad填充位上或者界外(无法取到有效值)时,舍弃该滑窗结果,在上述推导公式基础上对应空间轴shape需减去1:

      {Hout=Hout1if (Hout1)sh>=Hin+padding_sizeHtopWout=Wout1if (Wout1)sw>=Win+padding_sizeWleft\begin{cases} H_{out}=H_{out} - 1& \text{if } (H_{out}-1)*s_h>=H_{in}+padding\_size_{Htop} \\ W_{out}=W_{out} - 1& \text{if } (W_{out}-1)*s_w>=W_{in}+padding\_size_{Wleft} \\ \end{cases}\\
[object Object]

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

[object Object]
[object Object]
[object Object]
  • 参数说明:

    [object Object]

    [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:gradOutput和self数据类型支持FLOAT。数据格式支持NCHW和CHW。indices数据类型支持INT32。仅支持dilation为(1,1)。

  • 返回值:

    aclnnStatus: 返回状态码,具体参见。 第一段接口完成入参校验,出现以下场景时报错:

    [object Object]
[object Object]
  • 参数说明:

    [object Object]
  • 返回值:

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

[object Object]
  • 确定性计算:

    • aclnnMaxPool2dWithIndicesBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
  • 输入数据暂不支持NaN、-Inf。indices值不能超过公式中的H_inW_inH\_in*W\_in,且需要大于等于0。

[object Object]

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

[object Object]