在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。（源代码））

我没有任何经验。如果速度对您很重要，那么您可能会需要为特定的应用程序进行一些挖掘和基准测试不同的代码。