Hello! This seems like it would be a huge boost for me, but I need support for complex numbers - any chance I could make a PR? Right now I'm getting:
a = tensor([[[9.9567e-01, 9.9567e-01, 9.9567e-01, ..., 9.9567e-01,
9.9567e-01, 9.9567e-01],
[9.9567e-0... [1.6774e-05, 1.6774e-05, 1.6774e-05, ..., 1.6774e-05,
1.6774e-05, 1.6774e-05]]], dtype=torch.float64)
b = tensor([[[ 0.0000+0.0000j, -0.0024-0.0698j, -0.0097-0.1392j, ...,
-0.0219+0.2079j, -0.0097+0.1392j, -0.0024...698j, -0.0097-0.1392j, ...,
-0.0219+0.2079j, -0.0097+0.1392j, -0.0024+0.0698j]]],
requires_grad=True)
def multiply_in_place(a, b):
> a.mul_(b)
E RuntimeError: result type ComplexDouble can't be cast to the desired output type Double