tensorflow - 为什么 DeepLabv3+ 的 predict_labels() 中没有 softmax？

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#L191

我期待的是：

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

那么为什么没有softmax呢？