reference/generated/torchrl.render.video.compose_frame_grid

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

compose_frame_grid

torchrl.render.video.compose_frame_grid(frames: Sequence[ndarray], layout: Literal['single', 'grid', 'horizontal', 'vertical'] = 'grid') → ndarray

Composes multiple frames into one RGB image.

Parameters:

• frames – Frames to compose.

• layout – "grid" (and "single") tile the frames into a near-square grid, "horizontal" composes one row, and "vertical" composes one column.