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,
... ):

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