Category: Data movement operations
Short description: VariadicSplit operation splits an input tensor into pieces along some axis. The pieces may have variadic lengths depending on *"split_lengths*" attribute.
No attributes available.
data- A tensor of type T1. Required.
axis- An axis along
datato split. A scalar of type T2 with value from range
-rank(data) .. rank(data)-1. Negative values address dimensions from the end. Required.
split_lengths- A list containing the sizes of each output tensor along the split
axis. Size of
split_lengthsshould be equal to the number of outputs. The sum of sizes must match
data.shape[axis]. A 1-D Tensor of type T2.
split_lenghtscan contain a single
-1element, that means all remaining items along specified
axisthat are not consumed by other parts. Required.
datatensor. The shape of the i-th output has the same shape as the
dataexcept along dimension
axiswhere the size is
split_lengths[i] != -1.
-1item, if exists, is processed as described in the
VariadicSplit operation splits the
data input tensor into pieces along
axis. The i-th shape of output tensor will be equal to the
data shape except along dimension
axis where the size will be
split_lengths[i]. The sum of elements of split_lengths must match
Shape of output tensor will be:
\[ shape_output_tensor = shape_input_tensor[shape_input_tensor, shape_input_tensor, ..., split_lengths[axis], ..., shape_input_tensor[D-1]], where D rank of input tensor. \]