pytorch下对简单的数据进行分类(classification)