mindspore.dataset.MappableDataset.add_sampler

查看源文件
mindspore.dataset.MappableDataset.add_sampler(new_sampler)[源代码]

为当前数据集添加子采样器。

参数:
  • new_sampler (Sampler) - 待添加的子采样器。

样例:

>>> import mindspore.dataset as ds
>>> dataset = ds.GeneratorDataset([i for i in range(10)], "column1")
>>>
>>> new_sampler = ds.DistributedSampler(10, 2)
>>> dataset.add_sampler(new_sampler)