batchnorm is a critical building block in modern convolutional neural networks. Its unique property of operating on "batches" instead of i