code: def np_split_squeeze(array, axis):
      axis_len = array.shape[axis]
      return [
          np.squeeze(arr, axis=(axis,))
          for arr in np.split(array, axis_len, axis=axis)
      ]
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import gradient_checker_v2
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling a global function with example usage: np_split_squeeze(array, axis) and returns: unknown