Функция reduce_max

Функция reduce_max возвращает тензор с максимальными значениями элементов исходного тензора, вычисленными по указанным осям. Первым параметром функция принимает исходный тензор. Вторым параметром передается ось или список осей, по которым выполняется редукция. Третьим параметром можно указать булево значение keepdims, которое сохраняет размерность исходного тензора.

Синтаксис

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

Пример

Давайте найдем максимальное значение среди всех элементов одномерного тензора:

import tensorflow as tf

t = tf.constant([1, 2, 3, 4, 5])
res = tf.reduce_max(t)
print(res)

Результат выполнения кода:

tf.Tensor(5, shape=(), dtype=int32)

Пример

Давайте найдем максимальные значения по столбцам двумерного тензора, то есть по оси с индексом 0:

<+python+> import tensorflow as tf t = tf.constant([[1, 2, 3], [4, 5, 6]]) res = tf.reduce_max(t, axis=0) print(res) <-python+>

Результат выполнения кода:

tf.Tensor([4 5 6], shape=(3,), dtype=int32)

Пример

Давайте найдем максимальные значения по строкам двумерного тензора, то есть по оси с индексом 1:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.reduce_max(t, axis=1)
print(res)

Результат выполнения кода:

tf.Tensor([3 6], shape=(2,), dtype=int32)

Пример

Давайте сохраним размерность исходного тензора с помощью параметра keepdims:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.reduce_max(t, axis=1, keepdims=True)
print(res)

Результат выполнения кода:

tf.Tensor(
[[3]
 [6]], shape=(2, 1), dtype=int32)

Смотрите также

функцию reduce_min,
которая вычисляет минимальное значение элементов тензора
функцию reduce_sum,
которая вычисляет сумму элементов тензора
функцию reduce_mean,
которая вычисляет среднее значение элементов тензора
функцию argmax,
которая возвращает индекс максимального элемента