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 | Целое число или список целых чисел длиной 1, N или N+2. Размер окна для каждого измерения входного тензора. |
strides | Целое число или список целых чисел длиной 1, N или N+2. Шаг скользящего окна для каждого измерения входного тензора. |
padding | Строка, либо 'VALID', либо 'SAME'. Алгоритм заполнения. См. здесь для получения дополнительной информации. |
data_format | Строка. Указывает измерение канала. Для N=1 может быть "NWC" (по умолчанию) или "NCW", для N=2 — "NHWC" (по умолчанию) или "NCHW", а для N=3 — "NDHWC" (по умолчанию) или "NCDHW". |
name | Необязательное имя операции. |
| Возвращаемое значение | |
|---|---|
Тензор формата, указанного в 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