tf.keras.losses.KLD

Computes Kullback-Leibler divergence loss between y_true and y_pred.