Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.common.log_probability

View source on GitHub

Computes log probability of actions given distribution.

tf_agents.utils.common.log_probability(
    distributions, actions, action_spec
)

Args:

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

Returns:

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