v-geler / fft-conv-pytorch Goto Github PK
View Code? Open in Web Editor NEWThis project forked from fkodom/fft-conv-pytorch
Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
License: MIT License