快捷方式

StorageEnsemble

class torchrl.data.replay_buffers.StorageEnsemble(*storages: Storage, transforms: List['Transform'] = None)[source]

儲存器的集合。

此類旨在與 ReplayBufferEnsemble 配合使用。

引數:

storages (Storage 序列) – 用於構建複合儲存器的儲存器列表。

關鍵字引數:

transforms (`Transform` 列表, 可選) – 與 storages 長度相同的 transforms 列表。

警告

此類的 get() 方法簽名與其他儲存器不匹配,因為它將返回一個元組 (buffer_id, samples) 而不是僅僅返回 samples。

警告

此類不支援寫入(類似於 WriterEnsemble)。要擴充套件其中一個回放緩衝區,只需索引父級 ReplayBufferEnsemble 物件即可。

attach(buffer: Any) None

此函式將 sampler 附加到此儲存器。

從此儲存器讀取的緩衝區必須透過呼叫此方法作為附加實體包含在內。這確保了當儲存器中的資料發生變化時,即使儲存器與其他緩衝區(例如 Priority Samplers)共享,元件也能感知到變化。

引數:

buffer – 從此儲存器讀取的物件。

dump(*args, **kwargs)

dumps() 的別名。

load(*args, **kwargs)

loads() 的別名。

save(*args, **kwargs)

dumps() 的別名。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

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

檢視教程

資源

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

檢視資源