Docstring should mention that you're also doing mean across channels. And are you sure mean across channels is the right thing?