ray.rllib.models.modelv2.ModelV2.variables

ModelV2.variables(as_dict: bool = False) → List[numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor] | Dict[str, numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor]

返回此模型的变量列表（或字典）。

参数:

as_dict – 变量是否应作为字典值返回（使用描述性的字符串键）。

返回:

此 ModelV2 的所有变量的列表（如果 as_dict 为 True，则为字典）。