tf.nn.avg_pool
| Просмотреть исходный код на GitHub |
Выполняет усредняющее пулинг на входе.
tf.nn.avg_pool(
input, ksize, strides, padding, data_format=None, name=None
)
Каждое значение в output является средним значением соответствующего окна размером ksize в value.
| Аргументы | |
|---|---|
input | Tensor ранга 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 | Необязательное имя для операции. |
| Возвращаемое значение | |
|---|---|
A 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/versions/r2.9/api_docs/python/tf/nn/avg_pool
