parameters: (n_a, n_b, name=None)
docstring: Calculate the batched KL divergence KL(n_a || n_b) with n_a and n_b Normal.
  
    Args:
      n_a: instance of a Normal distribution object.
      n_b: instance of a Normal distribution object.
      name: (optional) Name to use for created operations.
        default is "kl_normal_normal".
  
    Returns:
      Batchwise KL(n_a || n_b)
docstring_summary: Calculate the batched KL divergence KL(n_a || n_b) with n_a and n_b Normal.
docstring_tokens: Calculate the batched KL divergence KL ( n_a || n_b ) with n_a and n_b Normal .
function: def _kl_normal_normal(n_a, n_b, name=None):
    """Calculate the batched KL divergence KL(n_a || n_b) with n_a and n_b Normal.
  
    Args:
      n_a: instance of a Normal distribution object.
      n_b: instance of a Normal distribution object.
      name: (optional) Name to use for created operations.
        default is "kl_normal_normal".
  
    Returns:
      Batchwise KL(n_a || n_b)
    """
    with ops.op_scope([n_a.mu, n_b.mu], name, "kl_normal_normal"):
      one = constant_op.constant(1, dtype=n_a.dtype)
      two = constant_op.constant(2, dtype=n_a.dtype)
      half = constant_op.constant(0.5, dtype=n_a.dtype)
      s_a_squared = math_ops.square(n_a.sigma)
      s_b_squared = math_ops.square(n_b.sigma)
      ratio = s_a_squared / s_b_squared
      return (math_ops.square(n_a.mu - n_b.mu) / (two * s_b_squared)
              + half * (ratio - one - math_ops.log(ratio)))
function_tokens: def _kl_normal_normal ( n_a , n_b , name = None ) : with ops . op_scope ( [ n_a . mu , n_b . mu ] , name , "kl_normal_normal" ) : one = constant_op . constant ( 1 , dtype = n_a . dtype ) two = constant_op . constant ( 2 , dtype = n_a . dtype ) half = constant_op . constant ( 0.5 , dtype = n_a . dtype ) s_a_squared = math_ops . square ( n_a . sigma ) s_b_squared = math_ops . square ( n_b . sigma ) ratio = s_a_squared / s_b_squared return ( math_ops . square ( n_a . mu - n_b . mu ) / ( two * s_b_squared ) + half * ( ratio - one - math_ops . log ( ratio ) ) )