tf.compat.v2.reduce_logsumexp

Вычисляет log(sum(exp(элементы по измерениям тензора))).

tf.compat.v2.reduce_logsumexp(
    input_tensor, axis=None, keepdims=False, name=None
)

Сводит input_tensor по измерениям, указанным в axis. Если keepdims не равно true, ранг тензора уменьшается на 1 для каждого элемента в axis. Если keepdims равно true, уменьшенные измерения сохраняются с длиной 1.

Если у axis нет элементов, все измерения сводятся, и возвращается тензор с одним элементом.

Эта функция более устойчива с точки зрения чисел, чем log(sum(exp(input))). Она предотвращает переполнение, вызванное взятием exp от больших входных данных, и подпотоки, вызванные взятием log от малых входных данных.

Например:

x = tf.constant([[0., 0., 0.], [0., 0., 0.]])
tf.reduce_logsumexp(x)  # log(6)
tf.reduce_logsumexp(x, 0)  # [log(2), log(2), log(2)]
tf.reduce_logsumexp(x, 1)  # [log(3), log(3)]
tf.reduce_logsumexp(x, 1, keepdims=True)  # [[log(3)], [log(3)]]
tf.reduce_logsumexp(x, [0, 1])  # log(6)