code: def approximate_split(x, num_splits, axis=0):
    
    size = shape_list(x)[axis]
    size_splits = [tf.div(size + i, num_splits) for i in range(num_splits)]
    return tf.split(x, size_splits, axis=axis)
docs: Split approximately equally into num_splits parts.
  
    Args:
      x: a Tensor
      num_splits: an integer
      axis: an integer.
  
    Returns:
      a list of num_splits Tensors.