PyTorch 中使用 DataLoader 生成分类仿真数据

使用 DataLoader 可以将生成的数据集加载到模型中进行训练。以下是如何使用 DataLoader 的示例代码：

from torch.utils.data import DataLoader

batch_size = 32

data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=True)

其中，batch_size 表示每个 batch 的大小，shuffle 表示是否打乱数据集顺序。使用 DataLoader 将数据集加载到模型中，可以方便地进行批量训练。