How much faster new module fft pytorch then numpy.fft ? Сould you give me some exact data(graph)? Something like this: if i use numpy.fft in this task i will spend x time , but if i use pytorch.fft …
You could profile your workload via e.g. torch.utils.benchmark and compare different shapes of the PyTorch implementation vs. the numpy one.