View source on GitHub |
Gets the CommonFields.LOG_PROBABILITY from info depending on type.
tf_agents.trajectories.policy_step.get_log_probability(
info: tf_agents.typing.types.NestedTensorOrArray
,
default_log_probability: Optional[types.Float] = None
) -> tf_agents.typing.types.Float