In this groupnorm, the norm seems to be applied over time dims too, ie. it is not fully time factorised