GitHub

tf.keras.ops.split

Split a tensor into chunks.

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

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.

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

A list of tensors.