tf.raw_ops.FractionalAvgPoolGrad
Вычисляет градиент функции FractionalAvgPool.
tf.raw_ops.FractionalAvgPoolGrad(
orig_input_tensor_shape,
out_backprop,
row_pooling_sequence,
col_pooling_sequence,
overlapping=False,
name=None
)
В отличие от FractionalMaxPoolGrad, нам не нужно находить arg_max для FractionalAvgPoolGrad, нам просто нужно равномерно обработать каждый элемент out_backprop в те индексы, которые образуют один и тот же пулинг-ячейку. Поэтому нам нужно знать только форму исходного тензора, а не весь тензор.
| Аргументы | |
|---|---|
orig_input_tensor_shape | A Tensor типа int64. Форма исходного тензора входных данных для fractional_avg_pool |
out_backprop | A Tensor. Должен быть одним из следующих типов: float32, float64, int32, int64. 4-мерный с формой [batch, height, width, channels]. Градиенты по отношению к выходу fractional_avg_pool. |
row_pooling_sequence | A Tensor типа int64. Последовательность пулинга по строкам, формирует область пулинга вместе с последовательностью пулинга по столбцам. |
col_pooling_sequence | A Tensor типа int64. Последовательность пулинга по столбцам, формирует область пулинга вместе с последовательностью пулинга по строкам. |
overlapping | Необязательный bool. По умолчанию False. При значении True, при пулинге значения на границе соседних ячеек пулинга используются обеими ячейками. Например:
Если последовательность пулинга — [0, 2, 4], то 16 в индексе 2 будет использоваться дважды. Результатом будет [41/3, 26/3] для фракционного усредняющего пулинга. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и out_backprop. |
© 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/raw_ops/FractionalAvgPoolGrad