initializing on the right device and with the right data type

Author: alexhagen

fft_conv_pytorch/fft_conv.py

@@ -84,7 +84,7 @@ def fft_conv(
     dilation_ = to_ntuple(dilation, n=n)
 
     # internal dilation offsets
-    offset = torch.zeros(1, 1, *dilation_).to(signal.device)
+    offset = torch.zeros(1, 1, *dilation_, device=signal.device, dtype=signal.dtype)
     offset[(slice(None), slice(None), *((0,) * n))] = 1.0
 
     # correct the kernel by cutting off unwanted dilation trailing zeros