SamplerEnsemble

class torchrl.data.replay_buffers.SamplerEnsemble(*samplers, p=None, sample_from_all=False, num_buffer_sampled=None)

一个采样器集合。

此类旨在与 ReplayBufferEnsemble 一起使用。它包含采样器以及采样策略的超参数。

参数:

samplers (Sampler 序列) – 用于构成复合采样器的采样器。

关键字参数:

• p (概率列表或张量, 可选) – 如果提供，则指示采样期间每个数据集的权重。

• sample_from_all (bool, 可选) – 如果为 True，则将从每个数据集进行采样。这与 p 参数不兼容。默认为 False。

• num_buffer_sampled (int, 可选) – 要采样的缓冲区数量。如果 sample_from_all=True，则此参数无效，因为它默认为缓冲区的数量。如果 sample_from_all=False，将根据概率 p 对缓冲区进行采样。

警告

info 字典中提供的索引被放置在带有 index 和 buffer_ids 键的 TensorDict 中，允许上层的 ReplayBufferEnsemble 和 StorageEnsemble 对象检索数据。此格式与只返回常规张量索引的其他采样器不同。