Why have a model output raw logits instead of softmax?

2

I think it’s because that logit.argmax() equals to softmax(logit).argmax() during inference.