Класс GlobalMaxPooling2D

Класс GlobalMaxPooling2D применяется к данным и выполняет операцию глобального максимального пулинга. Слой находит максимальное значение для каждого канала входного тензора и возвращает одно число на канал. Первым параметром можно передать размер входных данных, вторым - формат данных.

Синтаксис

tf.keras.layers.GlobalMaxPooling2D(data_format=None)

Пример

Давайте создадим слой глобального максимального пулинга и применим его к тензору с формой 1, 2, 3, 2:

import tensorflow as tf
import numpy as np

layer = tf.keras.layers.GlobalMaxPooling2D()
t = tf.constant([[[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12]]]])
res = layer(t)
print(res)

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

tf.Tensor([[11 12]], shape=(1, 2), dtype=int32)

Пример

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

import tensorflow as tf

layer = tf.keras.layers.GlobalMaxPooling2D()
t = tf.constant([[[[1, 5], [2, 6]], [[3, 7], [4, 8]]]])
res = layer(t)
print(res)

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

tf.Tensor([[4 8]], shape=(1, 2), dtype=int32)

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

слой MaxPooling2D,
который выполняет максимальный пулинг над двумерными данными
слой GlobalAveragePooling2D,
который выполняет глобальный средний пулинг
слой GlobalMaxPooling1D,
который выполняет глобальный максимальный пулинг для одномерных данных
слой Flatten,
который преобразует многомерный тензор в одномерный