src/transformers/models/ernie_m/modeling_ernie_m.py/ErnieMAttention/__init__
class ErnieMAttention:    def __init__(self, config, position_embedding_type=None):
        super().__init__()
        self.self_attn = ErnieMSelfAttention(config, position_embedding_type=position_embedding_type)
        self.out_proj = nn.Linear(config.hidden_size, config.hidden_size)
        self.pruned_heads = set()