Pytorch多分类

继二分类 后，通过学习 nn.CrossEntropyLoss() 、softmax() 、argmax() ，我尝试实现了下多分类模型基本工作流程。本博文主要用于整理个人的知识框架，希望也能帮到大家。如有不足，欢迎留言。🙏