我是 PyTorch 的新手。奇怪的是，我找不到与此相关的任何内容，尽管它看起来很简单。
我想用特定的示例来构建我的批次，例如每批次的所有示例都具有相同的标签，或者只是用仅 2 个类的示例填充批次。
我该怎么做呢？对我来说，这似乎是数据加载器中的正确位置，而不是数据集中的正确位置？由于数据加载器负责批次而不是数据集？
有一个简单的最小示例吗？
pytorch pytorch-dataloader
pytorch ×1
pytorch-dataloader ×1