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