RoundRobinWriter

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

一个用于可组合经验回放缓冲区的 RoundRobin Writer 类。

参数:

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

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

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

extend(data: Sequence) → Tensor

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