在C中实现2D卷积的快速方法

#include<stdio.h>
#include <stdlib.h>

void conv2d(float input[][][], float kernel[][][], float output[][][], int in_rows, int in_cols, int k_rows, int k_cols) {
    int out_rows = in_rows - k_rows + 1;
    int out_cols = in_cols - k_cols + 1;

    for (int i = 0; i < out_rows; i++) {
        for (int j = 0; j < out_cols; j++) {
            float sum = 0;
            for (int k = 0; k < k_rows; k++) {
                for (int l = 0; l < k_cols; l++) {
                    sum += input[i + k][j + l] * kernel[k][l];
                }
            }
            output[i][j] = sum;
        }
    }
}

int main() {
    int in_rows = 4, in_cols = 4;
    int k_rows = 2, k_cols = 2;
    float input[4][4] = {
        {1, 2, 3, 4},
        {5, 6, 7, 8},
        {9, 10, 11, 12},
        {13, 14, 15, 16}
    };
    float kernel[2][2] = {
        {1, 2},
        {3, 4}
    };
    float output[3][3];

    conv2d(input, kernel, output, in_rows, in_cols, k_rows, k_cols);

    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 3; j++) {
            printf("%.2f ", output[i][j]);
        }
        printf("\n");
    }

    return 0;
}