Rate this Page

torchrl.trainers.algorithms.configs.transforms.FlattenObservationConfig#

class torchrl.trainers.algorithms.configs.transforms.FlattenObservationConfig(in_keys: list[str] | None = None, out_keys: list[str] | None = None, _target_: str = 'torchrl.envs.transforms.transforms.FlattenObservation')[source]#

Configuration for FlattenObservation transform.