code: def split(x, split_dim, num_or_size_splits, name=None):
    
    return SplitOperation(x, split_dim, num_or_size_splits, name=name).outputs
docs: Like tf.split.
  
    Args:
      x: a Tensor
      split_dim: a Dimension in x.shape.dims
      num_or_size_splits: either an integer dividing split_dim.size
         or a list of integers adding up to split_dim.size
      name: an optional string
    Returns:
      a list of Tensors.