CANN/ops-nn三维平均池化反向传播算子

AvgPool3DGrad

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn

产品支持情况

功能说明

• 算子功能：三维平均池化的反向传播，计算三维平均池化正向传播的输入梯度。

• 计算公式：

$$ D_{out} = \left\lfloor \frac{D_{in} + 2 \times \text{pads}[0] - \text{ksize}[0]}{\text{strides}[0]} + 1 \right\rfloor $$

$$ H_{out} = \left\lfloor \frac{H_{in} + 2 \times \text{pads}[1] - \text{ksize}[1]}{\text{strides}[1]} + 1 \right\rfloor $$

$$ W_{out} = \left\lfloor \frac{W_{in} + 2 \times \text{pads}[2] - \text{ksize}[2]}{\text{strides}[2]} + 1 \right\rfloor $$

若属性ceil_mode为True，且满足条件$(D_{out} - 1) \times \text{stride}[0] \geq D_{in} + \text{padding}[0]$，则会跳过最后一个窗口，这将导致维度$D_{out}$减少 1。此规则同样适用于维度$W_{out}$和$H_{out}$。

参数说明

约束说明

无。

调用说明

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn