Yes, that’s correct. Because softmax is a monotonic function, argmax will give you the same answer either with the “logits” or the softmax outputs as the input to argmax. The point is that they almost always define TF models with linear outputs (logits) at the output layer and then use from_logits = True mode in the cost function (the appropriate version of cross entropy), so that the activation calculation is done internally by the cost function.