reference/generated/torchrl.data.replay_buffers.WriterEnsemble

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

WriterEnsemble

class torchrl.data.replay_buffers.WriterEnsemble(*writers)

An ensemble of writers.

This class is designed to work with ReplayBufferEnsemble. It contains the writers but blocks writing with any of them.

Parameters:

writers (sequence of Writer) – the writers to make the composite writer.

Warning

This class does not support writing. To extend one of the replay buffers, simply index the parent ReplayBufferEnsemble object.

add()

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

extend()

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