ListStorage¶
- class torchrl.data.replay_buffers.ListStorage(max_size: int)[源代码]¶
存储在列表中的存储。
此类不能使用 PyTrees 扩展,在调用
extend()
期间提供的数据应该是可迭代的(如列表、元组、张量或具有非空批大小的张量字典)。- 参数::
max_size (int) – 存储中存储的元素的最大数量。
- attach(buffer: Any) None ¶
此函数将采样器附加到此存储。
从该存储读取的缓冲区必须通过调用此方法作为附加实体包含。这保证了当存储中的数据发生变化时,即使存储与其他缓冲区共享(例如,优先级采样器),组件也会意识到这些变化。
- 参数::
buffer – 从该存储读取的对象。