Pytorch with complex tensors

Is there a solution for dealing with complex tensors yet? Is there a similar function to numpy.angle in PyTorch?



You can check this issue to see the internal work on complex tensors:

Also third party libraries might give some partial support for complex tensors. A quick search points to for example.

thanks! I just found a way around complex number with some certain precision loss.