Can this deal with complex numbers?

[Ni-Chen]

Hi authors,

I intended to use this method on complex numbers and it turned out with a error message like:

File "optimizer.py", line 433, in <listcomp> * torch.randint_like( RuntimeError: check_random_bounds handles only integral, floating-point and boolean types

I'm wondering if it's possible to improve this for complex numbers? Thanks.

Ni