|View source on GitHub|
Reduces the input tensor along the given axis using Kahan summation.
tfp.substrates.jax.math.reduce_kahan_sum( input_tensor, axis=None, keepdims=False, name=None )
Returns both the total and the correction term, as a
representing the sum in higher precision as
total - correction.
A practical use-case is computing the difference of two large (magnitude) sums
we expect to be nearly equal. If instead we take their difference as
(s0.total - s1.total) - (s0.correction - s1.correction), we can retain more
precision in computing their difference.
total holds all the high-order bits of the sum, so the correction
can be safely neglected if further enhanced precision computations are not