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

_, out_height, out_width = self.output_shape()

X = X.reshape(batch_size*channels, 1, height, width)

X_col = image_to_column(X, self.pool_shape, self.stride, self.padding)

MaxPool or AveragePool specific method

output = self._pool_forward(X_col)

output = output.reshape(out_height, out_width, batch_size, channels)

output = output.transpose(2, 3, 0, 1)

return output

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_shap