将 matlab FFT 转换为 CUDA FFT

Question

我目前正在使用以下 matlab 函数：

function out = fft_2d(in)

out = fftshift(fft2(ifftshift(in)));

据我了解，这需要一个“自然顺序”输入，并将其“交换”到 fft2，然后使用 fftshift 再次移动 fft2 的结果以返回给我自然排序输出。这是正确的吗？

我正在将此代码移植到 C，并且我想使用 CUFFT 来执行此操作。根据文档，我想我会使用：

/* Create a 2D FFT plan. */
cufftPlan2d(&plan, NX, NY, CUFFT_C2R);
/* Use the CUFFT plan to transform the signal out of place. */
cufftExecC2R(plan, idata, odata);

但是我必须对来自 cufftExecC2R 的数据进行什么样的转换？另外，odata 是否需要是 NX*NY 连续数据块？是否必须按列或行主要顺序？我猜，因为那就是C。

谢谢

Answer 1

CUDA FFT 的输入：

由于频域格式复杂，焊盘宽度到单元 (宽度/2+1)*2。
该初始填充将是结果图像的大小 -->必须作物结果。

然后填充到 2 的整数次方 * 2 实数浮点矩阵的整数次方（从右侧和底部填充零）。

输出，r,i,r,i,...甚至复杂的浮点值（实数列，虚数列，实数列，...），在中心周围填充零。

在频域中使用复数乘法，而不是常规乘法。

IFFT 后，裁剪图像的边以获得高度 * ceil(宽度/2+1)*2 中心。
再次裁剪以删除右侧可能存在的多余线条（如果有的话）（裁剪为高度*宽度）。

不要忘记FIT-shift。我记不清具体是什么时候，
因此，如果结果错误，请尝试在 ifft snd 之后移位，然后在 fft 之后移位。

尝试乘以 delta 内核进行测试。

内核应围绕中心而不是角落填充。

对于偶数矩阵，中心是中心右侧下方的半个单元格。

您可以编写自定义 CUDA 内核来进行填充，然后编写另一个内核来一次性进行取消填充和移位。