Theano中的逆FFT

我不能评论代码的健壮性,但Theano作为一个整体仍在开发中(版本0.7) sandbox 我认为,这应该被认为比Theano的其他地方更不可靠。

很明显,这个FFT操作是不完整的,因为它目前无法计算梯度(注意TODO注释)。如果你需要渐变,那么对不起,这个操作不会有帮助(也许你可以完成它并提交增强?)

这个实现只是numpy实现的一个垫片,所以如果numpy的实现足够完整和可靠,可以执行您想要的操作,那么这个Theano垫片可能也是如此。

请注意,因为这只是包装numpy,所以它不会在GPU上运行,如果您将此操作与其他支持GPU的操作混合在一起,并在GPU上执行,则由于数据在主内存和GPU内存之间来回复制,速度会变慢。

要使用此操作,请执行以下操作:

import theano
import theano.sandbox.fourier as tsf

tsf.ifft(frames=..., n=..., axis=...)