register_gym_spec_conversion

torchrl.envs.register_gym_spec_conversion(spec_type)

為特定 spec 型別註冊轉換函式的裝飾器。

該方法必須具有以下簽名

>>> @register_gym_spec_conversion("spec.name")
... def convert_specname(
...     spec,
...     dtype=None,
...     device=None,
...     categorical_action_encoding=None,
...     remap_state_to_observation=None,
...     batch_size=None,
... ):

其中 gym(nasium).spaces.spec.name 是 spec 在 gym 中的位置。

如果 spec 型別可訪問，這樣做也可以

>>> @register_gym_spec_conversion(SpecType)
... def convert_specname(
...     spec,
...     dtype=None,
...     device=None,
...     categorical_action_encoding=None,
...     remap_state_to_observation=None,
...     batch_size=None,
... ):

..note:: 包裝的函式可以簡化，未使用的 kwargs 可以包裝在 **kwargs 中。