Softmax and loss lab
Change logits and temperature. The target changes the loss, not the predicted probabilities.
Inputs
Try making the correct token's logit low. Watch its negative log-likelihood rise.
Predicted distribution
Entropy measures uncertainty in the model's distribution. Cross-entropy for a one-hot target is −ln(probability of the correct token).