src/transformers/models/nat/modeling_nat.py/NatEmbeddings/__init__
class NatEmbeddings:    def __init__(self, config):
        super().__init__()

        self.patch_embeddings = NatPatchEmbeddings(config)

        self.norm = nn.LayerNorm(config.embed_dim)
        self.dropout = nn.Dropout(config.hidden_dropout_prob)