How to convert FFT complex-number array of spectrogram into pytorch tensor?

Hi All,

I notice that PyTorch is unable to use complex tensor.

What will you do if you face a complex-number array when doing FFT?

Thank you very much!


The current solution is to use tensors of shape [*, 2] where * is the shape of your complex array. All our spectral ops (*fft, stft) expect this format.

We are also working on enabling complex support. Actually the first couple PRs just landed on master.

Thank you very much!

Hope it can be integrated in pytorch 1.0.

; )

@HotpotPandaXG @SimonW I am also confused the problem of complex computation in Pytorch. Have you fixed it?
Hope for your kind help!

Best regards.