取樣器集合¶
- 類 torchrl.data.replay_buffers.取樣器集合(*samplers, p=None, sample_from_all=False, num_buffer_sampled=None)[源]¶
一個取樣器集合。
此類旨在與
回放緩衝區集合配合使用。它包含取樣器以及取樣策略超引數。- 引數:
samplers (Sampler 序列) – 構成複合取樣器的取樣器。
- 關鍵字引數:
p (list 或機率張量,可選) – 如果提供,表示在取樣過程中每個資料集的權重。
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物件檢索資料。此格式與通常將索引作為常規張量返回的其他取樣器不同。