Add function to get the Kullback-Leibler divergence of the Gaussian distribution with respect to the standard normal distribution.
Add function to get the Kullback-Leibler divergence of the Gaussian distribution with respect to the standard normal distribution.

This is often used in variational inference, and useful as a regularizer.

PiperOrigin-RevId: 299315286

deepmind/rlax,deepmind/rlax