Phase Unwrapping

Hi,

I would like to know how to implement a phase unwrapping like “numpy.unwrap” with the backward functionality? Could it be possible to support “torch.stft” with magnitude and phase instead of real and imaginary components to make life easier?

Regards