python - 使用 CUDA 或 OpenCL 在 python 中进行多维 FFT

Question

我一直在浏览对 FFT 进行编程以在我的显卡上工作的简单方法（这是最近支持 CUDA 3.s​​omething 的 NVIDIA）。我目前的选择是学习 C，然后是 CUDA 的特殊 C 版本，或者使用一些 python CUDA 函数。我宁愿不学习 C，因为我只用高级语言编程。我查看了 pyCUDA 和其他在 python 中使用我的图形卡的方法，但我找不到任何只能与 python 代码一起使用的 FFT 库。一些库/项目似乎处理类似的项目（CUDAmat，Theano），但遗憾的是我没有发现 FFT。

numpy.fft.fft2()是否存在可以使用我的图形卡执行相同操作的功能？

编辑：开源解决方案的奖励点。

Answer 1

有PyFFT，它是开源的，基于 Apple 的（有些有限的）实现。免责声明：我在 PyFFT 上工作 :)

Answer 2

是的，ArrayFire 有一个用于 Python 的二维 FFT。

免责声明：我在 ArrayFire 上工作。