code: def map_decode_back_pieces(encoded_input, ori_tokens, tokenizer):
      decoded = [
          tokenizer.decode(x) for x in encoded_input["input_ids"]
      ]
      pieces = []
      ori_cnt = 0
      current = []
      for d in decoded:
          if d == "":
              continue
          current.append(d)
          if "".join(current) == ori_tokens[ori_cnt]:
              pieces.append(current)
              current = []
              ori_cnt += 1
      assert len(pieces) == len(ori_tokens)
      return pieces
  
imports: import json
  from argparse import ArgumentParser
  from transformers import BertTokenizer, RobertaTokenizer, AutoTokenizer
label_desc: Use image node_id 2 for calling a global function with example usage: map_decode_back_pieces(encoded_input, ori_tokens, tokenizer) and returns: pieces