The batch norm layer