python - 非立方体メッシュでの FFT 速度

Question

微分方程式を解くために、3D 関数のフーリエ変換/逆フーリエ変換を繰り返し実行する必要があります。何かのようなもの：

import pyfftw.interfaces.numpy_fft as fftw
for i in range(largeNumber):
    fFS = fftw.rfftn(f)
    # Do stuff
    f = fftw.irfftn(fFS)

f の形状は非常に非立方体です。(512, 32, 128) と (512, 128, 32) など、次元の順序に基づいてパフォーマンスの違いはありますか?

利用可能なスピードアップを探しています。私はすでに知恵をいじってみました。fFS.shape = (32, 128, 257) になるように、最大​​の次元が最後 (たとえば 32, 128, 512) になると最速になると思いましたが、そうではないようです。