|---| | [ View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.15.1/tensorflow/python/ops/numpy_ops/np_array_ops.py#L1152-L1158) | TensorFlow variant of NumPy's `split`. ```python tf.experimental.numpy.split( ary, indices_or_sections, axis=0 ) ``` See the NumPy documentation for [`numpy.split`](https://numpy.org/doc/stable/reference/generated/numpy.split.html).