fix batchnorm dimension issue
#4
by prathikr - opened
According to the ONNX Spec for BatchNormalization, there is no explicitly defined behavior for input of rank 2. There is precedent for input of rank 1 where we can assume a 2nd dimension of size 1. Extending that precedent, this change adds an unsqueeze node before all BatchNormalization operators to add a 3rd dimension of size 1 and then squeezes away that dimension for downstream operators.
➡️
