def METHOD_NAME(self, src_tokens):
    # embed tokens and positions
    x = embed = self.embed_scale * self.embed_tokens(src_tokens)
    if self.embed_positions is not None:
        x = embed + self.embed_positions(src_tokens)
    # contextualize embeddings
    x = x.transpose(0, 1)
    x = self.dropout_module(x)
    x, _ = self.emb_ctx.forward(x)
    x = x.transpose(0, 1)
    if self.layernorm_embedding is not None:
        x = self.layernorm_embedding(x)
    x = self.dropout_module(x)
    return x, embed