Thanks @ptrblck for your reply, from this conversation I came to know that softmax is used only for calculating the CrossEntropyLoss  rather than classification
Can I print or get the softmax output which is used in CrossEntropyLoss just for curiosity