pytorch
diff --git a/‎torchtext/vocab.py
Copy file name to clipboardExpand all lines: torchtext/vocab.py
+6Lines changed: 6 additions & 0 deletions b/‎torchtext/vocab.py
Copy file name to clipboardExpand all lines: torchtext/vocab.py
+6Lines changed: 6 additions & 0 deletions
@@ -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))