Helper function for Vocab object

torchtext/vocab.py

@@ -103,6 +103,12 @@ def __init__(self, counter, max_size=None, min_freq=1, specials=('<unk>', '<pad>
     def _default_unk_index(self):
         return self.unk_index
 
+    def get_token(self, index):
+        result = self.itos[index]
+        if result is None:
+            return Vocab.UNK
+        return result
+
     def __getitem__(self, token):
         return self.stoi.get(token, self.stoi.get(Vocab.UNK))