Function prepare_sequence

beginner_source/nlp/advanced_tutorial.py:136–138

(seq, to_ix)

def prepare_sequence(seq, to_ix):
    idxs = [to_ix[w] for w in seq]
    return torch.tensor(idxs, dtype=torch.long)


# Compute log sum exp in a numerically stable way for the forward algorithm