I have an RNN network model that is connected to a linear layer. My question is that if I use the softmax function outside the model(like outside the forward function) to calculate loss with cross-entropy loss, will it be included in the loss.backward() calculation?