reference/generated/torchrl.render.collect_render_rollouts

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

collect_render_rollouts

torchrl.render.collect_render_rollouts(env: Any, policy: Any, config: RenderConfig) → RenderResult

Collects sequential render rollouts.

EnvBase environments are rolled out through rollout(); other environments fall back to a duck-typed reset/step loop. Captured frames span the initial state through the terminal state of each trajectory.

Parameters:

• env – Environment returned by torchrl.render.make_render_env().

• policy – TensorDict-compatible policy.

• config – Render configuration.

Returns:

A render result containing trajectories and in-memory frames.