megengine.data.ReplacementSampler

class ReplacementSampler(dataset, batch_size=1, drop_last=False, num_samples=None, weights=None, world_size=None, rank=None, seed=None)[源代码]

基类:megengine.data.sampler.MapSampler

有放回(可重复选取)地进行随机采样。

参数

weights (List) – 样本索引的权重,可以是未归一化的权重。

方法

__init__(dataset[, batch_size, drop_last, …])

初始化方法。

batch()

Batch 方法提供了一个 batch 索引生成器。

sample()

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

scatter(indices)

分散方法用于将下标拆分为子集,每个子集将分配一个编号。