Fix avgpooling forwardims when kernel size > input dim size (!382) · Merge requests · Eclipse Projects / aidge / aidge_core

Noam Zerah requested to merge noamzerah/aidge_core:fix_avgpooling_forwardims into dev Apr 01, 2025

This MR fixes dimension calculation in AvgPool when kernel size exceeds input dimensions by forcing output=1 when kernel is larger than input (matching PyTorch/TensorFlow behavior). This problem was the cause of [#261 (closed)]

Edited Apr 01, 2025 by Noam Zerah

Fix avgpooling forwardims when kernel size > input dim size

Merge request reports