ray.rllib.core.rl_module.rl_module.RLModule.输出规格_推理

RLModule.output_specs_inference() → List[str | Tuple[str, ...]]

返回 forward_inference() 方法的输出规格。

重写此方法以自定义推理调用的输出规格。默认实现要求 forward_inference() 方法返回一个字典，该字典包含 action_dist 键，其值是 Distribution 的一个实例。