We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4c2374d commit 2a59602Copy full SHA for 2a59602
fft_conv_pytorch/fft_conv.py
@@ -21,7 +21,7 @@ def complex_matmul(a: Tensor, b: Tensor, groups: int = 1) -> Tensor:
21
22
# complex value matrix multiplication
23
real = a.real @ b.real - a.imag @ b.imag
24
- imag = a.imag @ b.real - a.real @ b.imag
+ imag = a.imag @ b.real + a.real @ b.imag
25
real = torch.movedim(real, real.dim() - 1, 2).squeeze(-1)
26
imag = torch.movedim(imag, imag.dim() - 1, 2).squeeze(-1)
27
c = torch.zeros(real.shape, dtype=torch.complex64, device=a.device)
0 commit comments