BatchSampler

    批采样器的基础实现,用于 paddle.io.DataLoader 中迭代式获取mini-batch的样本下标数组,数组长度与 batch_size 一致。

    所有用于 paddle.io.DataLoader 中的批采样器都必须是 的子类并实现以下方法:

    __iter__: 迭代式返回批样本下标数组。

    __len__: 每epoch中mini-batch数。

    • dataset (Dataset) - 此参数必须是 paddle.io.Dataset 或 的一个子类实例或实现了 __len__ 的Python对象,用于生成样本下标。默认值为None。

    • shuffle (bool) - 是否需要在生成样本下标时打乱顺序。默认值为False。

    paddle.io.DataLoader

    返回:返回样本下标数组的迭代器。

    返回类型: BatchSampler

    代码示例