code: def _insert_back_keepdims(x, axis):
      """Insert the dims in `axis` back as singletons after being removed.
  
      Args:
        x:  `Tensor`.
        axis:  Python list of integers.
  
      Returns:
        `Tensor` with same values as `x`, but additional singleton dimensions.
      """
      for i in sorted(axis):
          x = tf.expand_dims(x, axis=i)
      return x
  
file_path: probability/tensorflow_probability/python/stats/quantiles.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import assert_util
  from tensorflow_probability.python.internal import distribution_util
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensorshape_util
label_desc: Use image node_id 8 for calling a global function with example usage: _insert_back_keepdims(x, axis) and returns: x