tf_agents.trajectories.policy_step.get_log_probability

Gets the CommonFields.LOG_PROBABILITY from info depending on type.