layer_kl_divergence_add_loss function

Pass-through layer that adds a KL divergence penalty to the model loss