How to align sub words to words in Sentencepiece like Albert for NER?

Hi @sgugger @lysandre , how to align subwords back to words in task like Token Classification NER etc .