Hi,
Many operations do not seem to be implemented for the dtype=complex yet. I am currently facing this problem:
import numpy as np
import torch
import torch.fft

x = torch.tensor(np.random.random((30, 20, 10)), requires_grad=True)
t = torch.fft.rfft(x, dim=2)
t = torch.prod(t, dim=1)  # or torch.exp(torch.sum(torch.log(t), dim=1))
xn = torch.fft.irfft(t, dim=1)

which gives RuntimeError: prod does not support automatic differentiation for outputs with complex dtype. Using torch.log results in similar problems.
Do you guys have any idea how to get around this problem for the above situation?
Thank you in advance.