Concise Implementation of Softmax Regression

https://d2l.ai/chapter_linear-classification/softmax-regression-concise.html