tf.keras.layers.GlobalMaxPool2D
| Просмотреть исходный код на GitHub |
Операция глобального максимального пулинга для пространственных данных.
tf.keras.layers.GlobalMaxPool2D(
data_format=None, keepdims=False, **kwargs
)
Примеры:
input_shape = (2, 4, 5, 3) x = tf.random.normal(input_shape) y = tf.keras.layers.GlobalMaxPool2D()(x) print(y.shape) (2, 3)
| Аргументы | |
|---|---|
data_format | Строка, одна из channels_last (по умолчанию) или channels_first. Порядок измерений в входных данных. channels_last соответствует входным данным с формой (batch, height, width, channels), а channels_first соответствует входным данным с формой (batch, channels, height, width). По умолчанию используется значение image_data_format из файла конфигурации Keras по адресу ~/.keras/keras.json. Если его никогда не устанавливали, то будет "channels_last". |
keepdims | Булево значение, сохранять ли пространственные измерения. Если keepdims равно False (по умолчанию), ранг тензора уменьшается для пространственных измерений. Если keepdims равно True, пространственные измерения сохраняются с длиной 1. Поведение аналогично tf.reduce_max или np.max. |
Форма входных данных:
- Если
data_format='channels_last': 4D тензор с формой(batch_size, rows, cols, channels). - Если
data_format='channels_first': 4D тензор с формой(batch_size, channels, rows, cols).
Форма выходных данных:
- Если
keepdims=False: 2D тензор с формой(batch_size, channels). - Если
keepdims=True:- Если
data_format='channels_last': 4D тензор с формой(batch_size, 1, 1, channels) - Если
data_format='channels_first': 4D тензор с формой(batch_size, channels, 1, 1)
- Если
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/GlobalMaxPool2D