code: def expand_dims(inputs, axis):
      """Expand dims on sparse, ragged, or dense tensors."""
      if isinstance(inputs, tf.SparseTensor):
          return tf.sparse.expand_dims(inputs, axis)
      else:
          return tf.expand_dims(inputs, axis)
  
imports: from keras.utils.module_utils import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: expand_dims(inputs, axis) and returns: tf, tf