src/transformers/models/beit/modeling_beit.py/BeitPooler/__init__
class BeitPooler:    def __init__(self, config):
        super().__init__()
        self.layernorm = (
            nn.LayerNorm(config.hidden_size, eps=config.layer_norm_eps) if config.use_mean_pooling else None
        )