tf.keras.ops.split

Split a tensor into chunks.

Main aliases

tf.keras.ops.numpy.split

tf.keras.ops.split(
    x, indices_or_sections, axis=0
)

Args
`x`	Input tensor.
`indices_or_sections`	If an integer, N, the tensor will be split into N equal sections along `axis`. If a 1-D array of sorted integers, the entries indicate indices at which the tensor will be split along `axis`.
`axis`	Axis along which to split. Defaults to `0`.

Note
A split does not have to result in equal division when using Torch backend.

Returns
A list of tensors.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/split