I am new to pytorch and while going through the MNIST example i saw that in the last layer we had provided no activation in the forward function . Would there be any difference if i add a softmax activation function at the output layer?