def _insert_back_keep_dims(x, axis):
for i in sorted(axis):<EOL><INDENT>x = tf.expand_dims(x, axis=i)<EOL><DEDENT>return x<EOL>
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.