A pre-canned builder for the analytic kl divergence.

def analytic_kl_builder(posterior, prior, sample):
  """A pre-canned builder for the analytic kl divergence."""
  del sample
  return tf.reduce_sum(tfp.distributions.kl_divergence(posterior, prior))