输入tensor的推导公式:
当ceilMode=False时,indices tensor的shape中H和W维度推导公式:
当ceilMode=True时,out tensor的shape中H和W维度推导公式:
如果滑窗左上角的起始位置处在下方或右侧的填充区域,或位于图像之外(无法获取有效值)时,则该滑窗结果会被舍弃,需要在上述推导公式的基础上,将对应空间轴的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。数据格式支持ND和NCHW,输入3维时表示CHW取值。indices数据类型支持INT32。仅支持dilation为(1,1)。
返回值:
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
确定性计算:
- aclnnMaxPool2dWithIndicesBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
输入数据暂不支持NaN、-Inf。indices值不能超过公式中的,且需要大于等于0。
[object Object]