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) – 转换前的规范
- 返回：
转换后预期的规范
