快捷方式

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) – 轉換前的規範

返回:

轉換後預期的規範

文件

獲取全面的 PyTorch 開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源