ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.utils.common.log_probability

Computes log probability of actions given distribution.

distributions A possibly batched tuple of distributions.
actions A possibly batched action tuple.
action_spec A nested tuple representing the action spec.

A Tensor representing the log probability of each action in the batch.