TensorDictRoundRobinWriter

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

用於可組合、基於 tensordict 的重播緩衝區的 RoundRobin Writer 類別。

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

在適當的索引處插入一個數據片段，並傳回該索引。

extend(data: Sequence) → Tensor

在適當的索引處插入一系列數據點，並傳回包含索引的張量。