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!