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 (布尔值, 可选) – 如果为 True，则将从每个数据集中进行采样。这与 p 参数不兼容。默认为 False。

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

警告

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