Skip to content

GlobalAveragePooling should not squeeze its reduced dimensions

What commit version of aidge do you use

  • aidge_core: 0.3.1

Problem description

GlobalAvergaePooling Operator should not squeeze dimensions it has just reduced if we follow the ONNX behaviour

The first two dimensions of output shape are the same as the input (N x C), while the other dimensions are all 1.

Removing these dimensions creates issues with imported models because of uncompatible dimensions.