快捷方式

混洗

class torchdata.dataloader2.adapter.Shuffle(enable=True)

Shuffle DataPipes 适配器允许控制图中所有现有的 Shuffler (shuffle) DataPipes。

参数::

enable

可选的布尔参数,用于在 DataPipe 图中启用/禁用混洗。默认情况下为 True。

  • True:启用之前禁用的所有 ShufflerDataPipes。如果不存在,它将在图的末尾添加一个新的 shuffle

  • False:禁用图中的所有 ShufflerDataPipes

  • None:无操作。为向后兼容性而引入。

示例

dp = IterableWrapper(range(size)).shuffle()
dl = DataLoader2(dp, [Shuffle(False)])
assert list(range(size)) == list(dl)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源