快捷方式

ListStorage

class torchrl.data.replay_buffers.ListStorage(max_size: int)[源代码]

存储在列表中的存储。

此类不能使用 PyTrees 扩展,在调用 extend() 期间提供的数据应该是可迭代的(如列表、元组、张量或具有非空批大小的张量字典)。

参数::

max_size (int) – 存储中存储的元素的最大数量。

attach(buffer: Any) None

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

从该存储读取的缓冲区必须通过调用此方法作为附加实体包含。这保证了当存储中的数据发生变化时,即使存储与其他缓冲区共享(例如,优先级采样器),组件也会意识到这些变化。

参数::

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

dump(*args, **kwargs)

dumps() 的别名。

load(*args, **kwargs)

loads() 的别名。

save(*args, **kwargs)

dumps() 的别名。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源