多线程并行计算卷积的C++代码

以下是一个基于OpenMP的多线程并行计算卷积的C++代码：

#include <iostream>
#include <vector>
#include <omp.h>

using namespace std;

vector<vector<double>> calculate_convolution(const vector<vector<double>>& input, const vector<vector<double>>& kernel)
{
    int input_height = input.size();
    int input_width = input[0].size();
    int kernel_height = kernel.size();
    int kernel_width = kernel[0].size();

    int output_height = input_height - kernel_height + 1;
    int output_width = input_width - kernel_width + 1;

    vector<vector<double>> output(output_height, vector<double>(output_width));

#pragma omp parallel for
    for (int i = 0; i < output_height; i++)
    {
        for (int j = 0; j < output_width; j++)
        {
            double sum = 0.0;
            for (int k = 0; k < kernel_height; k++)
            {
                for (int l = 0; l < kernel_width; l++)
                {
                    sum += input[i + k][j + l] * kernel[k][l];
                }
            }
            output[i][j] = sum;
        }
    }

    return output;
}

int main()
{
    vector<vector<double>> input = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}};
    vector<vector<double>> kernel = {{1, 1}, {1, 1}};

    vector<vector<double>> output = calculate_convolution(input, kernel);

    for (int i = 0; i < output.size(); i++)
    {
        for (int j = 0; j < output[0].size(); j++)
        {
            cout << output[i][j] << " ";
        }
        cout << endl;
    }

    return 0;
}

在这个例子中，我们使用了OpenMP库来实现并行计算卷积。通过使用`#pragma omp parallel for`指令，我们将卷积的计算任务分配给了多个线程。在这个例子中，我们假设输入和卷积核都是方形的，但可以通过修改代码来适应不同形状的输入和卷积核。