SamplerEnsemble

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

取樣器集合。

此類旨在與 ReplayBufferEnsemble 配合使用。它包含取樣器以及取樣策略的超引數。

引數:

samplers (sequence of Sampler) – 用於構建複合取樣器的取樣器。

關鍵字引數:

• p (list 或 tensor of probabilities, optional) – 如果提供，則表示取樣過程中每個資料集的權重。

• sample_from_all (bool, optional) – 如果為 True，則將從每個資料集中進行取樣。這與 p 引數不相容。預設為 False。

• num_buffer_sampled (int, optional) – 要取樣的緩衝區數量。如果 sample_from_all=True，則此引數無效，因為它預設為緩衝區數量。如果 sample_from_all=False，將根據機率 p 對緩衝區進行取樣。

警告

資訊字典中提供的索引被放置在 TensorDict 中，鍵為 index 和 buffer_ids，允許上層的 ReplayBufferEnsemble 和 StorageEnsemble 物件檢索資料。此格式與其他取樣器不同，後者通常將索引作為常規張量返回。