快捷方式

ListStorage

class torchrl.data.replay_buffers.ListStorage(max_size: int | None = None, compilable: bool = False)[source]

基于列表的存储。

此类不能通过 PyTrees 扩展,在调用 extend() 时提供的数据应为可迭代对象(如列表、元组、张量或具有非空批量大小的 tensordicts)。

参数:

max_size (int, optional) – 存储中存储的最大元素数量。如果未提供,则创建一个无限大小的存储。

attach(buffer: Any) None

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

读取此存储的缓冲区必须通过调用此方法作为附加实体包含在内。这确保了当存储中的数据发生更改时,即使存储与其他缓冲区(例如优先采样器)共享,组件也能感知到更改。

参数:

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

dump(*args, **kwargs)

dumps() 的别名。

load(*args, **kwargs)

loads() 的别名。

save(*args, **kwargs)

dumps() 的别名。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深度教程

查看教程

资源

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

查看资源