Pytorch with complex tensors

1

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

thanks!

2

Hi,

You can check this issue to see the internal work on complex tensors: https://github.com/pytorch/pytorch/issues/755

Also third party libraries might give some partial support for complex tensors. A quick search points to https://github.com/williamFalcon/pytorch-complex-tensor for example.

3

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