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 物件檢索資料。此格式與通常將索引作為常規張量傳回的其他取樣器不同。