TensorDictRoundRobinWriter

class torchrl.data.replay_buffers.TensorDictRoundRobinWriter(**kw)

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

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

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

extend(data: Sequence) → Tensor

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