megengine.data.MapSampler¶

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

基类：megengine.data.sampler.Sampler

字典数据集的采样器。

参数

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

方法

`__init__`(dataset[, batch_size, drop_last, …])	初始化方法。
`batch`()	Batch 方法提供了一个 batch 索引生成器。
`sample`()	返回一个列表，包含了所有样本的索引。
`scatter`(indices)	分散方法用于将下标拆分为子集，每个子集将分配一个编号。