updated conv.py with changing imag variable

This commit is contained in:
Izam Mohammed 2023-01-24 07:12:45 +05:30 committed by GitHub
parent 379800d3f6
commit 40031ab084
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -129,7 +129,7 @@ class ComplexConvTranspose2d(nn.Module):
imag_real = self.real_conv(imag)
real = real_real - imag_imag
imag = real_imag - imag_real
imag = real_imag + imag_real
out = torch.cat([real, imag], 1)