tf.contrib.layers.avg_pool2d
Добавляет операцию 2D усредняющего пулинга.
tf.contrib.layers.avg_pool2d(
inputs, kernel_size, stride=2, padding='VALID', data_format=DATA_FORMAT_NHWC,
outputs_collections=None, scope=None
)
Предполагается, что пулинг выполняется по каждой изображению, но не по партиям или каналам.
| Аргументы | |
|---|---|
inputs | Четырёхмерный тензор формы [batch_size, height, width, channels] если data_format равно NHWC, и [batch_size, channels, height, width] если data_format равно NCHW. |
kernel_size | Список длины 2: [высота_ядра, ширина_ядра] ядра пулинга, по которому вычисляется операция. Может быть целым числом, если оба значения одинаковы. |
stride | Список длины 2: [шаг_по_высоте, шаг_по_ширине]. Может быть целым числом, если оба шага одинаковы. Обратите внимание, что в настоящее время оба шага должны иметь одинаковое значение. |
padding | Метод заполнения, либо 'VALID', либо 'SAME'. |
data_format | Строка. NHWC (по умолчанию) и NCHW поддерживаются. |
outputs_collections | Коллекции, в которые добавляются результаты. |
scope | Необязательный scope для name_scope. |
| Возвращает | |
|---|---|
A Tensor представляющий результат операции пулинга. |
| Возможные исключения | |
|---|---|
ValueError | Если data_format не равно ни NHWC, ни NCHW. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/layers/avg_pool2d