Versioned name: VariadicSplit-1
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.
Attributes
No attributes available.
Inputs
data
- A tensor of type T1. Required.axis
- An axis along data
to 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_lengths
should be equal to the number of outputs. The sum of sizes must match data.shape[axis]
. A 1-D Tensor of type T2. split_lenghts
can contain a single -1
element, that means all remining items along specified axis
that are not consumed by other parts. Required.Outputs
data
tensor. The shape of the i-th output has the same shape as the data
except along dimension axis
where the size is split_lengths[i]
if split_lengths[i] != -1
. -1
item, if exists, is processed as described in the split_lengths
input description.Detailed Description
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 data.shape[axis]
.
Shape of output tensor will be:
Types
Examples