Function make_context_vector

beginner_source/nlp/word_embeddings_tutorial.py:347–349

(context, word_to_ix)

def make_context_vector(context, word_to_ix):
    idxs = [word_to_ix[w] for w in context]
    return torch.tensor(idxs, dtype=torch.long)


make_context_vector(data[0][0], word_to_ix)  # example