Replaces the composition of fft1D into one fft2D

rishikksh20 / FNet-pytorch

Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms

MIT License

251 stars 37 forks source link

Open AdilZouitine opened 3 years ago

AdilZouitine commented 3 years ago

Hi, I propose you a slight factorisation of the code and it is clearer to read.