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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-06-07 UTC.

Args