如何使用CUFFT的批处理模式？

#include <cufft.h>

cufftHandle plan;
cufftResult result;
result = cufftPlan1d(&plan, n, CUFFT_C2C, batch);

cufftComplex *input, *output;
cudaMalloc(&input, sizeof(cufftComplex) * n * batch);
cudaMalloc(&output, sizeof(cufftComplex) * n * batch);

// 将输入数据分批复制到输入数组中
for (int i = 0; i< batch; i++) {
    cudaMemcpy(input + i * n, input_data[i], sizeof(cufftComplex) * n, cudaMemcpyHostToDevice);
}

cufftExecC2C(plan, input, output, CUFFT_FORWARD);

// 将输出数据分批复制到输出数据数组中
for (int i = 0; i< batch; i++) {
    cudaMemcpy(output_data[i], output + i * n, sizeof(cufftComplex) * n, cudaMemcpyDeviceToHost);
}

cufftDestroy(plan);
cudaFree(input);
cudaFree(output);