GrayScale¶
- class torchrl.envs.transforms.GrayScale(in_keys: Sequence[NestedKey] | None = None, out_keys: Sequence[NestedKey] | None = None)[source]¶
將畫素觀測值轉換為灰度圖。
- transform_observation_spec(observation_spec: TensorSpec) TensorSpec[source]¶
轉換觀測規範,使轉換後的規範與轉換對映匹配。
- 引數:
observation_spec (TensorSpec) – 轉換前的規範
- 返回:
轉換後預期的規範