Spec-Zone.ru › TensorFlow 1.15

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API