快捷方式

取樣器集合

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中,鍵為indexbuffer_ids,允許上層的ReplayBufferEnsembleStorageEnsemble物件檢索資料。此格式與通常將索引作為常規張量返回的其他取樣器不同。

文件

查閱 PyTorch 的綜合開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源