Method to

bert/tokenization_utils_base.py:522–525

Send all values to device by calling v.to(device)

(self, device: str)

    @torch_required
    def to(self, device: str):
        """Send all values to device by calling v.to(device)"""
        self.data = {k: v.to(device) for k, v in self.data.items()}
        return self


# class AddedToken(UserString):