ENH : add support for equalize_channels in AverageTFR