ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Splits input tensor across all dimensions.

An op which slices the input tensor based on the given num_splits attribute, pads slices optionally, and returned the slices. Slices are returned in row-major order.

This op may be generated via the TPU bridge.

For example, with input tensor:

[[0, 1, 2],
 [3, 4, 5],
 [6, 7, 8]]


[2, 2]

and paddings:

[1, 1]

the expected outputs is:

[[0, 1],
 [3, 4]]
[[2, 0],
 [5, 0]]
[[6, 7],
 [0, 0]]
[[8, 0],
 [0, 0]]

input A Tensor. Input tensor to split across all dimensions. } out_arg { name: "outputs" description: <
N An int that is >= 1.
num_splits A list of ints. Number of ways to split per dimension. Shape dimensions must be evenly divisible.
paddings An optional list of ints. Defaults to []. Optional list of right paddings per dimension of input tensor to apply before splitting. This can be used to make a dimension evenly divisible.
name A name for the operation (optional).

A list of N Tensor objects with the same type as input.