tf.contrib.layers.avg_pool3d
Добавляет операцию 3D усредняющего пулинга.
tf.contrib.layers.avg_pool3d(
inputs, kernel_size, stride=2, padding='VALID', data_format=DATA_FORMAT_NDHWC,
outputs_collections=None, scope=None
)
Предполагается, что пулинг выполняется по изображению, но не по пакету или каналам.
| Аргументы | |
|---|---|
inputs | 5-мерный тензор формы [batch_size, depth, height, width, channels] если data_format равно NDHWC, и [batch_size, channels, depth, height, width] если data_format равно NCDHW. |
kernel_size | Список длиной 3: [kernel_depth, kernel_height, kernel_width] ядра пулинга, по которому вычисляется операция. Может быть целым числом, если все значения одинаковые. |
stride | Список длиной 3: [stride_depth, stride_height, stride_width]. Может быть целым числом, если все шаги одинаковые. Обратите внимание, что в настоящее время оба шага должны иметь одинаковое значение. |
padding | Способ заполнения, либо 'VALID', либо 'SAME'. |
data_format | Строка. NDHWC (по умолчанию) и NCDHW поддерживаются. |
outputs_collections | Коллекции, в которые добавляются результаты. |
scope | Необязательный scope для name_scope. |
| Возвращаемые значения | |
|---|---|
A Tensor представляющий результаты операции пулинга. |
| Исключения | |
|---|---|
ValueError | Если data_format не равно ни NDHWC, ни NCDHW. |
© 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_pool3d
