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 中。