pytorch函数分类