Softmax回归的简洁实现

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