- 输入tensor的推导公式:
当ceilMode=False时,indices tensor的shape中H和W维度推导公式:
当ceilMode=True时,out tensor的shape中H和W维度推导公式:
滑窗左上角起始位处在下或右侧pad填充位上或者界外(无法取到有效值)时,舍弃该滑窗结果,在上述推导公式基础上对应空间轴shape需减去1:
每个算子分为,必须先调用“aclnnMaxPool2dWithIndicesBackwardGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnMaxPool2dWithIndicesBackward”接口执行计算。
[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)。
返回值:
[object Object]
确定性计算:
- aclnnMaxPool2dWithIndicesBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
输入数据暂不支持NaN、-Inf。indices值不能超过公式中的,且需要大于等于0。
[object Object]