Numpy实现MaxPooling2D（最大池化）和AveragePooling2D（平均池化）(4)

def backward_pass(self, accum_grad):

batch_size, _, _, _ = accum_grad.shape

channels, height, width = self.input_shape

accum_grad = accum_grad.transpose(2, 3, 0, 1).ravel()

MaxPool or AveragePool specific method

accum_grad_col = self._pool_backward(accum_grad)

accum_grad = column_to_image(accum_grad_col, (batch_size * channels, 1, height, width), self.pool_shape, self.stride, 0)

accum_grad = accum_grad.reshape((batch_size,) + self.input_shape)

return accum_grad

def output_shape(self):

channels, height, width = self.input_shape

out_height = (height - self.pool_shape[0]) / self.stride + 1

out_width = (width - self.pool_shape[1]) / self.stride + 1

assert out_height % 1 == 0

assert out_width % 1 == 0

return channels, int(out_height), int(out_width)

class MaxPooling2D(PoolingLayer):

def _pool_forward(self, X_col):

arg_max = np.argmax(X_col, axis=0).flatten()

output = X_col[arg_max, range(arg_max.size)]

self.cache = arg_max

return output

def _pool_backward(self, accum_grad):

accum_grad_col = np.zeros((np.prod(self.pool_shape), accum_grad.size))

arg_max = self.cache

accum_grad_col[arg_max, range(accum_grad.size)] = accum_grad

return accum_grad_col

class AveragePooling2D(PoolingLayer):

def _pool_forward(self, X_col):

output = np.mean(X_col, axis=0)

return output

def _pool_backward(self, accum_grad):

accum_grad_col = np.zeros((np.prod(self.pool_shape), accum_grad.size))

accum_grad_col[:, range(accum_grad.size)] = 1. / accum_grad_col.shape[0] * accum_grad

return accum_grad_col

[

超强干货来袭
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Python工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Python开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上前端开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以扫码获取！！！（备注：Python）

知识点，真正体系化！**

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以扫码获取！！！（备注：Python）