文件最后提交记录最后更新时间
6 个月前
5 个月前
6 个月前
6 个月前
5 个月前
README

MaxPool3DGrad

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT √
Atlas A3 训练系列产品/Atlas A3 推理系列产品 ×
Atlas A2 训练系列产品/Atlas A2 推理系列产品 ×
Atlas 200I/500 A2推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×

功能说明

算子功能:正向最大池化后反向传播,将梯度回填到每个窗口最大值的坐标处,相同坐标处累加。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
orig_x 输入 待进行MaxPool3DGrad计算的入参,表示正向的输入Tensor。 FLOAT、FLOAT16、BFLOAT16 NCDHW
orig_y 输入 表示正向输入中最大元素的索引位置。数据格式需要与`orig_x`一致,shape需要与`grads`一致。 FLOAT、FLOAT16、BFLOAT16 NCDHW
grads 输入 待进行MaxPool3DGrad计算的入参,表示当前节点的梯度。 FLOAT、FLOAT16、BFLOAT16 NCDHW
ksize 属性 表示最大池化的窗口大小。 INT64 -
strides 属性 表示池化操作的步长。 INT64 -
padding 可选属性 pad模式,本次仅支持SAME和VALID,填补的数据为-inf,默认SAME。 - -
pads 属性 当pad模式为CALCULATED时,输入的orig_x的shape进行左右上下前后的扩展,填补的数据为-inf,暂不生效。 - -
data_format 可选属性 表示支持的数据格式。 - 本次仅支持NCDHW,后续支持NDHWC
y 输出 待进行MaxPool3DGrad计算的出参。shape、数据格式需要与`orig_x`一致。 FLOAT、FLOAT16、BFLOAT16 NCDHW

约束说明

无。

调用说明

不涉及。