reference/generated/torchrl.envs.transforms.GrayScale

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

GrayScale

class torchrl.envs.transforms.GrayScale(in_keys: Sequence[NestedKey] | None = None, out_keys: Sequence[NestedKey] | None = None)

Turns a pixel observation to grayscale.

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

Transforms the observation spec such that the resulting spec matches transform mapping.

Parameters:

observation_spec (TensorSpec) – spec before the transform

Returns:

expected spec after the transform