python - 提高 Python 中的 FFT 性能

Question

Python 中最快的 FFT 实现是什么？

似乎 numpy.fft 和 scipy.fftpack 都基于 fftpack，而不是 FFTW。 fftpack 和 FFTW 一样快吗？使用多线程 FFT 或分布式 (MPI) FFT 怎么样？

Answer 1

您当然可以使用 Cython 或其他允许您访问外部库的类似工具来封装您想要测试的任何 FFT 实现。

基于 GPU

如果您要测试 FFT 实现，您还可以查看基于 GPU 的代码(如果您可以使用适当的硬件)。有几个:reikna.fft , scikits.cuda .

基于 CPU 的

还有一个基于 CPU 的 python FFTW 包装器 pyFFTW .

(也有pyFFTW3，但不像pyFFTW那样积极维护，而且它不适用于Python3。(source))

我对这些都没有经验。如果速度对您很重要，您可能需要为您的特定应用程序进行一些挖掘和基准测试。