как преобразовать метод pytorch adaptive_avg_pool2d в керас или тензорный поток

Question

Я не знаю, как преобразовать метод PyTorch adaptive_avg_pool2d в Keras или TensorFlow.Кто-нибудь может помочь?Метод PyTorch:

adaptive_avg_pool2d(14,[14])

Я попытался использовать средний пул, изменил тензор в Keras, но получил ошибку:

ValueError: общий размер нового массива должен бытьбез изменений

Answer 1

Я не уверен, что понял ваш вопрос, но в PyTorch вы передаете пространственные измерения AdaptiveAvgPool2d.Например, если вы хотите иметь размер вывода 5x7, вы можете использовать nn.AdaptiveAvgPool2d((5,7)).

Если вы хотите получить средний средний уровень пула, вы можете использовать nn.AdaptiveAvgPool2d(1).В Keras вы можете просто использовать GlobalAveragePooling2D.

Для других выходных размеров в Keras вам нужно использовать AveragePooling2D, но вы не можете указатьФорма вывода напрямую.Вам необходимо вычислить / определить параметры pool_size, stride и padding в зависимости от того, как вы хотите получить выходную форму.Если вам нужна помощь с расчетами, проверьте эту страницу курса CS231n .