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