Network in Network (NiN)

http://d2l.ai/chapter_convolutional-modern/nin.html

in the NiN code, I did not see the output softmax function. Is this on purpose?

The loss function used in d2l.train_ch5 is gluon.loss.SoftmaxCrossEntropyLoss() which deals with softmax.