code: def remove_batch(shape):
      if any(
          isinstance(i, list) or isinstance(i, tuple) for i in shape
      ):  # multi shape
          return [remove_batch(s) for s in shape]
      else:
          return list(shape[1:])
  
imports: import glob
  from bigdl.dllib.utils.file_utils import Sample
  import numpy
  from pyspark.ml.linalg import DenseVector, SparseVector, VectorUDT
  import sys
  import os
  import re
  from bigdl.dllib.utils.log4Error import *
label_desc: Use image node_id 16 for calling a global function with example usage: remove_batch(shape) and returns: unknown, list