reference/generated/torchrl.data.replay_buffers.TensorDictRoundRobinWriter

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictRoundRobinWriter

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

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

add(data: Any) → int | 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.