embedding layer

model.py

class CNNModel(nn.Module):
    def __init__(self, embedding_dim, vocab_size):
        super().__init__()
        self._word_embeddings = nn.Embedding(vocab_size, embedding_dim)

    def forward(self, sample):
        embeds = self._word_embeddings(sample)