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

tf_agents.agents.ppo.ppo_utils.nested_kl_divergence

View source on GitHub

Given two nested distributions, sum the KL divergences of the leaves.

tf_agents.agents.ppo.ppo_utils.nested_kl_divergence(
    nested_from_distribution, nested_to_distribution, outer_dims=()
)