pytorch里面使用transform对数据进行预处理，使用random_split_partial实现对数据集的划分