实现过程前面几节都有都可以参考，不再赘述。
    化身调包侠即可。而且感觉 Pytorch 提供的函数在训练时数值更加稳定。

    3.7 softmax 回归的简洁实现.py