tf.compat.v1.layers.average_pooling2d
Слой усредняющего пулинга для 2D входов (например, изображений).
tf.compat.v1.layers.average_pooling2d(
inputs,
pool_size,
strides,
padding='valid',
data_format='channels_last',
name=None
)
Миграция на TF2
Этот API является устаревшим API, который совместим только с выполнением eager и tf.function, если вы объединяете его с tf.compat.v1.keras.utils.track_tf1_style_variables
Обратитесь к разделу отображения моделей tf.layers в руководстве по миграции, чтобы узнать, как использовать вашу TensorFlow v1 модель в TF2 с Keras.
Соответствующий слой TensorFlow v2 — tf.keras.layers.AveragePooling2D.
Структурное отображение в родной TF2
Ни одно из поддерживаемых аргументов не изменило своего названия.
До:
y = tf.compat.v1.layers.average_pooling2d(x, pool_size=2, strides=2)
После:
Для миграции кода, использующего функциональные слои TF1, используйте Функциональный API Keras:
x = tf.keras.Input((28, 28, 1)) y = tf.keras.layers.AveragePooling2D(pool_size=2, strides=2)(x) model = tf.keras.Model(x, y)
Описание
| Аргументы | |
|---|---|
inputs | Тензор, над которым выполняется пулинг. Должен иметь ранг 4. |
pool_size | Целое число или кортеж/список из 2 целых чисел: (pool_height, pool_width), определяющий размер окна пулинга. Может быть одним целым числом для задания одного значения для всех пространственных измерений. |
strides | Целое число или кортеж/список из 2 целых чисел, определяющие шаги операции пулинга. Может быть одним целым числом для задания одного значения для всех пространственных измерений. |
padding | Строка. Метод заполнения, либо 'valid', либо 'same'. Регистронезависимый. |
data_format | Строка. Порядок измерений на входах. channels_last (по умолчанию) и channels_first поддерживаются. channels_last соответствует входам с формой (batch, height, width, channels), в то время как channels_first соответствует входам с формой (batch, channels, height, width). |
name | Строка, имя слоя. |
| Возвращаемое значение | |
|---|---|
| Тензор вывода. |
| Исключения | |
|---|---|
ValueError | если выполнение eager включено. |
© 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/compat/v1/layers/average_pooling2d