tf_agents.utils.common.log_probability

Stay organized with collections Save and categorize content based on your preferences.

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.