|TensorFlow 1 version||View source on GitHub|
Compute the cumulative log-sum-exp of the tensor
Compat aliases for migration
See Migration guide for more details.
tf.math.cumulative_logsumexp( x, axis=0, exclusive=False, reverse=False, name=None )
By default, this op performs an inclusive cumulative log-sum-exp, which means that the first element of the input is identical to the first element of the output.
This operation is significantly more numerically stable than the equivalent
computes the same result given infinite numerical precision. However, note
that in some cases, it may be less stable than
for a given element, as it applies the "log-sum-exp trick" in a different
More precisely, where
tf.math.reduce_logsumexp uses the following trick:
log(sum(exp(x))) == log(sum(exp(x - max(x)))) + max(x)
it cannot be directly used here as there is no fast way of applying it
to each prefix
x[:i]. Instead, this function implements a prefix
scan using pairwise log-add-exp, which is a commutative and associative
(up to floating point precision) operator:
log_add_exp(x, y) = log(exp(x) + exp(y)) = log(1 + exp(min(x, y) - max(x, y))) + max(x, y)
However, reducing using the above operator leads to a different computation tree (logs are taken repeatedly instead of only at the end), and the maximum is only computed pairwise instead of over the entire prefix. In general, this leads to a different and slightly less precise computation.
Tensor. Must be one of the following types:
int64(default: 0). Must be in the range
True, perform exclusive cumulative log-sum-exp.
True, performs the cumulative log-sum-exp in the reverse direction.
name: A name for the operation (optional).
Tensor. Has the same shape and type as