TensorDictRoundRobinWriter

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

用於可組合的、基於 tensordict 的回放緩衝區的輪詢寫入器類。

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

將一條資料插入到適當的索引，並返回該索引。

extend(data: Sequence) → Tensor

將一系列資料點插入到適當的索引，並返回包含這些索引的張量。