I am fairly new to python and I\'m trying to figure out how to extract the mapped tokens in small_bert [https://hub.tensorflow.google.cn/tensorflow/bert_en_uncased_
small_bert