Jul 14, 2021
For multi class you'd use softmax since it ensures that the probability for each class sum up to 1 and therefore are a valid pdf.
Written by Paul Gavrikov
PhD student in Computer Vision working on Representation Learning in Convolutional Neural Networks