tf_agents.utils.common.log_probability

View source on GitHub

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.