def METHOD_NAME(self, inputs):
    mean = inputs[0]
    log_var = inputs[1]
    kl = 1.0 + log_var - math_ops.square(mean) - math_ops.exp(log_var)
    kl = -0.5 * math_ops.reduce_mean(kl, axis=-1, keepdims=True)
    return kl