But, would I not want the logits as one of the inputs to Cross Entropy loss?
Also, I am not trying to do binary classification, I have to predict among 3 classes
I think the real problem I am facing is interfacing the output of RNN to a linear layer
Do you think this is correct?
self.rnn = nn.RNN(input_size = input_size, hidden_size = hidden_size,num_layers= num_layers, 
                          batch_first= True)
        
self.out = nn.Linear(hidden_size, 3)