RoundRobinWriter

class torchrl.data.replay_buffers.RoundRobinWriter(compilable: bool = False)

用于可组合回放缓冲区的 RoundRobin Writer 类。

参数:

compilable (bool, optional) – 写入器是否可编译。如果为 True，则写入器不能在多个进程之间共享。默认为 False。

add(data: Any) → int | torch.Tensor

将一条数据插入到适当的索引，并返回该索引。

extend(data: Sequence) → Tensor

将一系列数据点插入到适当的索引，并返回包含这些索引的张量。