View source on GitHub |
Sets the CommonFields.LOG_PROBABILITY on info to be log_probability.
tf_agents.trajectories.policy_step.set_log_probability(
info: tf_agents.typing.types.NestedTensorOrArray
,
log_probability: tf_agents.typing.types.Float
) -> tf_agents.typing.types.NestedTensorOrArray