Почему нет softmax в предикате предиката () в DeepLabv3+?

Question

В функции

def predict_labels(images, model_options, image_pyramid=None)

из model.py мы имеем:

predictions[output] = tf.argmax(logits, 3)

https://github.com/tensorflow/models/blob/1c99681ed94495f446399e247a8d63e07f28391c/research/deeplab/model.py

Я ожидал что-то вроде:

predictions[output] = tf.argmax(tf.nn.softmax(logits), 3)

Так почему там нет софтмакса?