No, you don’t need to apply the softmax activation to get the predictions, since the logits would already give you the predicted class. Softmax would only normalize the values to probabilities and you would get the same answer:
x = torch.randn(10, 10)
print(torch.argmax(x, dim=1))
print(torch.argmax(F.softmax(x, dim=1), dim=1))