def METHOD_NAME(self):
    embedding = []
    for i in range(self.n_layers):
        if i == 0:
            embedding.append(nn.Linear(self.input_dim, self.d_hidden))
            embedding.append(nn.ReLU())
        else:
            embedding.append(nn.Linear(self.d_hidden, self.d_hidden))
            embedding.append(nn.ReLU())
    return nn.Sequential(*embedding)