Функция maximum
Функция maximum сравнивает два тензора поэлементно
и возвращает новый тензор, каждый элемент которого равен
наибольшему из соответствующих элементов входных тензоров.
Первым параметром передаётся первый тензор, вторым - второй тензор.
Оба тензора должны иметь одинаковую форму или быть broadcast-совместимыми.
Синтаксис
tf.maximum(x, y, [name])
Пример
Давайте найдём поэлементный максимум двух тензоров:
import tensorflow as tf
a = tf.constant([1, 2, 3, 4, 5])
b = tf.constant([5, 4, 3, 2, 1])
res = tf.maximum(a, b)
print(res)
Результат выполнения кода:
tf.Tensor([5 4 3 4 5], shape=(5,), dtype=int32)
Пример
Давайте найдём поэлементный максимум для двумерных тензоров:
import tensorflow as tf
a = tf.constant([[1, 2, 3], [4, 5, 6]])
b = tf.constant([[6, 5, 4], [3, 2, 1]])
res = tf.maximum(a, b)
print(res)
Результат выполнения кода:
tf.Tensor(
[[6 5 4]
[4 5 6]], shape=(2, 3), dtype=int32)
Пример
Давайте применим функцию к тензору и скаляру:
import tensorflow as tf
a = tf.constant([1, 2, 3, 4, 5])
res = tf.maximum(a, 3)
print(res)
Результат выполнения кода:
tf.Tensor([3 3 3 4 5], shape=(5,), dtype=int32)
Смотрите также
-
функцию
minimum,
которая возвращает поэлементный минимум двух тензоров -
функцию
reduce_max,
которая находит максимальное значение по осям тензора -
функцию
argmax,
которая возвращает индекс максимального значения -
функцию
clip_by_value,
которая ограничивает значения тензора диапазоном