reference/generated/torchrl.envs.transforms.AutoResetEnv

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

AutoResetEnv

class torchrl.envs.transforms.AutoResetEnv(*args, **kwargs)

A subclass for auto-resetting envs.

insert_transform(index: int, transform: Transform) → None

Inserts a transform to the env at the desired index.

Transform or callable are accepted.