Softmax回归的简洁实现

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