https://zh.d2l.ai/chapter_linear-networks/softmax-regression-scratch.html
本书第二版，pytorch版本的PDF，什么时候能改成pytorch的版本，现在下载下来的PDF仍然是mxnet