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
代码示例