tf.nn.avg_pool
Выполняет усредняющее пулинг-преобразование входных данных.
tf.nn.avg_pool(
input, ksize, strides, padding, data_format=None, name=None
)
Используется в блокнотах
| Используется в учебных руководствах |
|---|
Каждый элемент в output представляет собой среднее значение соответствующего окна ksize размера в value.
| Аргументы | |
|---|---|
input | Тензор ранга N+2 с формой [batch_size] + input_spatial_shape + [num_channels], если data_format не начинается с «NC» (по умолчанию), или [batch_size, num_channels] + input_spatial_shape, если data_format начинается с «NC». Пулинг выполняется только по пространственным измерениям. |
ksize | Целое число или список ints длины 1, N или N+2. Размер окна для каждого измерения входного тензора. |
strides | Целое число или список ints длины 1, N или N+2. Шаг скользящего окна для каждого измерения входного тензора. |
padding | Строка: 'VALID' или 'SAME'. Алгоритм дополнения. Подробнее см. здесь. |
data_format | Строка. Задает измерение каналов. При N=1 может иметь значение "NWC" (по умолчанию) или "NCW", при N=2 — "NHWC" (по умолчанию) или "NCHW", а при N=3 — "NDHWC" (по умолчанию) или "NCDHW". |
name | Необязательное имя операции. |
| Возвращает | |
|---|---|
Тензор Tensor формата, заданного параметром data_format. Тензор выходных данных, полученный усредняющим пулингом. |
© 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/api_docs/python/tf/nn/avg_pool