StorageEnsemble

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

存储的集合。

此类旨在与 ReplayBufferEnsemble 一起使用。

参数:

storages (Storage 序列) – 用于构成复合存储的存储。

关键字参数:

transforms (Transform 列表, 可选) – 与存储长度相同的转换列表。

警告

此类的 get() 签名与其他存储不匹配，因为它将返回一个元组 (buffer_id, samples) 而不仅仅是样本。

警告

此类不支持写入（类似于 WriterEnsemble）。要扩展其中一个回放缓冲区，只需索引父 ReplayBufferEnsemble 对象。

attach(buffer: Any) → None

此函数将采样器附加到此存储。

从该存储读取数据的缓冲区必须通过调用此方法作为附加实体包含在内。 这保证了当存储中的数据更改时，即使存储与其他缓冲区（例如，优先级采样器）共享，组件也会意识到更改。

参数:

buffer – 从该存储读取数据的对象。

dump(*args, **kwargs)

dumps() 的别名。

load(*args, **kwargs)

loads() 的别名。

save(*args, **kwargs)

dumps() 的别名。