src/transformers/data/datasets/language_modeling.py/TextDataset/__getitem__
class TextDataset:    def __getitem__(self, i) -> torch.Tensor:
        return torch.tensor(self.examples[i], dtype=torch.long)