reference/generated/torchrl.data.replay_buffers.Writer

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

Writer

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

A ReplayBuffer base Writer class.

abstract add(data: Any) → int

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

abstract extend(data: Sequence) → Tensor

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