tf_agents.agents.ppo.ppo_utils.nested_kl_divergence

View source on GitHub

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