// 定义池化操作 void pool(float *input, int input_width, int input_height, int width, int height, int stride, float *output, int output_width, int output_height) { for (int i = 0; i < output_height; i++) { for (int j = 0; j < output_width; j++) { float max_val = -INFINITY; for (int k = 0; k < height; k++) { for (int l = 0; l < width; l++) { int x = j * stride + l; int y = i * stride + k; if (x >= 0 && x < input_width && y >= 0 && y < input_height) { float val = input[y * input_width + x]; if (val > max_val) { max_val = val; } } } } output[i * output_width + j] = max_val; } } }改为平均池化

可以将池化操作的求最大值改为求平均值即可，具体实现如下：

// 定义平均池化操作
void avg_pool(float *input, int input_width, int input_height,
              int width, int height, int stride,
              float *output, int output_width, int output_height) {
    for (int i = 0; i < output_height; i++) {
        for (int j = 0; j < output_width; j++) {
            float sum_val = 0.0f;
            int count = 0;
            for (int k = 0; k < height; k++) {
                for (int l = 0; l < width; l++) {
                    int x = j * stride + l;
                    int y = i * stride + k;
                    if (x >= 0 && x < input_width && y >= 0 && y < input_height) {
                        float val = input[y * input_width + x];
                        sum_val += val;
                        count++;
                    }
                }
            }
            output[i * output_width + j] = sum_val / count;
        }
    }
}

在这个函数中，我们使用了一个 `sum_val` 变量来累加每个池化窗口内的像素值，然后再除以像素数来求平均值。注意，为了避免除以 0 的情况，我们在每个池化窗口内记录了像素数，并用变量 `count` 来累加。最后，将求得的平均值存储到输出数组中即可。