ObservationTransform¶
- class torchrl.envs.transforms.ObservationTransform(in_keys: collections.abc.Sequence[tensordict._nestedkey.NestedKey] | None = None, out_keys: collections.abc.Sequence[tensordict._nestedkey.NestedKey] | None = None, in_keys_inv: collections.abc.Sequence[tensordict._nestedkey.NestedKey] | None = None, out_keys_inv: collections.abc.Sequence[tensordict._nestedkey.NestedKey] | None = None)[source]¶
Abstract class for transformations of the observations.