tf.split

tf.split

tf.split

split(
    value,
    num_or_size_splits,
    axis=0,
    num=None,
    name='split'
)

Defined in tensorflow/python/ops/array_ops.py.

See the guide: Tensor Transformations > Slicing and Joining

Splits a tensor into sub tensors.

If num_or_size_splits is an integer type, num_split, then splits value along di