TensorDictRoundRobinWriter

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

A RoundRobin Writer class for composable, tensordict-based replay buffers.

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

Inserts one piece of data at an appropriate index, and returns that index.

extend(data: Sequence) → Tensor

Inserts a series of data points at appropriate indices, and returns a tensor containing the indices.