莫烦pytorch学习笔记（六）——区分类型（分类）