nip.utils.plotting.rollouts

nip.utils.plotting.rollouts#

Utilities for processing raw rollouts for use in plotting.

Functions

get_last_timestep_mask(rollouts)

Compute a mask for the last timestep of each rollout.