tf.keras.ops.average_pool
Операция усредняющего пулинга.
tf.keras.ops.average_pool(
inputs,
pool_size,
strides=None,
padding='valid',
data_format=None
)
| Аргументы | |
|---|---|
inputs | Тензор ранга N+2. inputs имеет форму (batch_size,) + inputs_spatial_shape + (num_channels,), если data_format="channels_last", или (batch_size, num_channels) + inputs_spatial_shape, если data_format="channels_first". Пулинг выполняется только по пространственным измерениям. |
pool_size | Целое число или кортеж/список целых чисел размера len(inputs_spatial_shape), задающий размер окна пулинга для каждого пространственного измерения входного тензора. Если pool_size — целое число, то для всех пространственных измерений используется одно и то же значение pool_size. |
strides | Целое число или кортеж/список целых чисел размера len(inputs_spatial_shape). Шаг скользящего окна для каждого пространственного измерения входного тензора. Если strides — целое число, то для всех пространственных измерений используется одно и то же значение strides. |
padding | Строка: либо "valid", либо "same". "valid" означает, что дополнение не применяется, а "same" — что дополнение равномерно добавляется слева/справа или сверху/снизу от входных данных, благодаря чему высота/ширина выходных данных совпадает с высотой/шириной входных данных, если strides=1. |
data_format | Строка: либо "channels_last", либо "channels_first". data_format определяет порядок измерений во входных данных. Если data_format="channels_last", inputs имеет форму (batch_size, ..., channels), а если data_format="channels_first", inputs имеет форму (batch_size, channels, ...). |
| Возвращает | |
|---|---|
| Тензор ранга N+2 — результат операции усредняющего пулинга. |
© 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/keras/ops/average_pool
