GrayScale¶
- class torchrl.envs.transforms.GrayScale(in_keys: Optional[Sequence[NestedKey]] = None, out_keys: Optional[Sequence[NestedKey]] = None)[source]¶
將像素觀察轉換為灰階。
- transform_observation_spec(observation_spec: TensorSpec) TensorSpec[source]¶
轉換觀察規格，使結果規格與轉換映射匹配。
- 參數:
observation_spec (TensorSpec) – 轉換前的規格
- 返回:
轉換後的預期規格
