hi @RalphMao .. fourier_feature is not wotk with pytorch 1.7.0 because the function torch.rfft is removed can you help my to edit your code to work with pytorch 1.7.0 thanks..... @Sibozhu .....