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

vision
1

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!

Best,
Panda

2

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.

3

Thank you very much!

Hope it can be integrated in pytorch 1.0.

; )

4

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

Best regards.