vae.ops.reduce_logmeanexp

vae.ops.reduce_logmeanexp(x, axis=None, keep_dims=False, name=None)

Compute log(sum(exp(x))) in a numerically stable manner

Parameters:

x : tf.Tensor

axis : int or list of int or None, optional (default=None)

A single, or a list of dimensions to reduce. If not provided, reduces all dimensions.

keep_dims : bool, optional (default=None)

If True, retains reduced dimensions with length one.

name : str or None, optional (default=None)

Name of the operation. Defaults to ‘ReduceLogMeanExp’.

Returns:

y : tf.Tensor