tf.math.reduce_max

Вычисляет tf.math.maximum элементов по всем измерениям тензора.

tf.math.reduce_max(
    input_tensor, axis=None, keepdims=False, name=None
)

Это операция сведения для элементной операции tf.math.maximum.

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

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

Пример использования:

x = tf.constant([5, 1, 2, 4])
tf.reduce_max(x)
<tf.Tensor: shape=(), dtype=int32, numpy=5>
x = tf.constant([-5, -1, -2, -4])
tf.reduce_max(x)
<tf.Tensor: shape=(), dtype=int32, numpy=-1>
x = tf.constant([4, float('nan')])
tf.reduce_max(x)
<tf.Tensor: shape=(), dtype=float32, numpy=nan>
x = tf.constant([float('nan'), float('nan')])
tf.reduce_max(x)
<tf.Tensor: shape=(), dtype=float32, numpy=nan>
x = tf.constant([float('-inf'), float('inf')])
tf.reduce_max(x)
<tf.Tensor: shape=(), dtype=float32, numpy=inf>

См. документацию numpy для np.amax и np.nanmax поведения.