Hi,
I’m trying to run a NN with complex tensors.
I keep getting the following error: RuntimeError: expected scalar type Float but found ComplexDouble
I suspect maybe it is because the weights are not complex?
If so- how do I change it?
> class NeuralNet(nn.Module):
>     def __init__(self, input_nodes, hidden1_nodes, hidden2_nodes, hidden3_nodes, output_nodes):
>         super(NeuralNet, self).__init__()
>         self.type(torch.complex128)
>         self.fc1 = nn.Linear(input_nodes, hidden1_nodes)
>         self.relu = nn.ReLU()
>         self.fc2 = nn.Linear(hidden1_nodes, hidden2_nodes)
>         self.fc3 = nn.Linear(hidden2_nodes, hidden3_nodes)
>         self.fc4 = nn.Linear(hidden3_nodes, output_nodes)

Thanks!