torch.utils
(1) torch.utils.data
torch.utils.data.random_split 数据集切分
dataset1, dataset2 = torch.utils.data.random_split(dataSetName,[55000,5000])
将数据集按比例划分,在装进dataLoader之前使用。
torch.utils.data.dataLoader 装进dataLoader
torch.utils.data.dataLoader(dataset=XXX,
batch_size=XXX,
shuffle=XXX)