- 接口功能:实现RoiPoolingWithArgMax的反向。遍历每个ROI的池化结果,将feature map坐标上的反向梯度贡献累加,即完成整张图上的反向计算。
- 计算公式:
其中,
判决函数
[object Object]表示i节点是否被候选区域r的第j个输出节点选为最大值输出
算子执行接口为,必须先调用“aclnnRoiPoolingGradWithArgMaxGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小,再调用“aclnnRoiPoolingGradWithArgMax”接口执行计算。
[object Object]
[object Object]
- gradOutput、rois、argmax 、gradInputRef的数据类型在支持的范围之内。
- gradOutput、argmax与gradInputRef具有相同的数据类型
- gradOutput、argmax、gradInputRef的shape大小为4,rois的shape大小为2
- gradOutput、argmax、rois的shape[0]相等
- gradOutput、argmax的shape[1]相等
- gradOutput、argmax的shape[2]等于pooledH和shape[3]等于pooledW
- rois的值大于等于0
- pooledH、pooledW大于0。
- rois[:, 1] 小于 rois[:, 2] 且 rois[:, 3] 小于 rois[:, 4]
- rois.shape[0]、gradOutput.shape[0]小于等于1024
[object Object]