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 字典中提供的索引放置在 TensorDict 中，其中键为 index 和 buffer_ids，这允许上层 ReplayBufferEnsemble 和 StorageEnsemble 对象检索数据。此格式与其他通常将索引作为常规张量返回的采样器不同。