fftshift можно использовать для переключения между соглашениями об индексах в fft.По сути, каждая реализация fft поставляется с ним, например, numpy , но в TensorFlow ее нет.
Как следует реализовать fftshift (и его обратное, ifftshift) в TensorFlow дифференцируемым способом