在Pytorch版本中生成模型的集合