RandomSampler¶

class RandomSampler(dataset, batch_size=1, drop_last=False, indices=None, world_size=None, rank=None, seed=None)[源代码]¶

不放回（不重复选取）地随机采样。

参数

dataset – 待采样的目标数据集。
batch_size – 使用 batch 方法时指定 batch 大小。
drop_last – 如果batch大小不能整除数据集大小时，为 True 则放弃最后一个不完整的batch; 为 False 则最后一个batch可能比较小。默认：False
indices – 样本的索引。
world_size – 进程数。
rank – 进程号，在0与 world_size 范围内的非负整数。
seed – 随机种子。

返回一个列表，包含了所有样本的索引。

SequentialSampler

ReplacementSampler