VariadicSplit#
Versioned name: VariadicSplit-1
Category: Data movement
Short description: VariadicSplit operation splits an input tensor into chunks along some axis. The chunks may have variadic lengths depending on split_lengths
input tensor.
Detailed Description
VariadicSplit operation splits a given input tensor data into chunks along a scalar or tensor with shape [1]
axis
. It produces multiple output tensors based on additional input tensor split_lengths
.
The i-th output tensor shape is equal to the input tensor data shape, except for dimension along axis which is split_lengths[i]
.
Where D is the rank of input tensor data. The sum of elements in split_lengths
must match data.shape[axis]
.
Attributes: VariadicSplit operation has no attributes.
Inputs
1:
data
. A tensor of type T1 and arbitrary shape. Required.2:
axis
. Axis alongdata
to split. A scalar or tensor with shape[1]
of typeT2
with value from range-rank(data) .. rank(data)-1
. Negative values address dimensions from the end. Required.3:
split_lengths
. A list containing the dimension values of each output tensor shape along the splitaxis
. A 1D tensor of typeT2
. The number of elements insplit_lengths
determines the number of outputs. The sum of elements insplit_lengths
must matchdata.shape[axis]
. In additionsplit_lengths
can contain a single-1
element, which means, all remaining items along specifiedaxis
that are not consumed by other parts. Required.
Outputs
Multiple outputs: Tensors of type
T1
. The i-th output has the same shape as data input tensor except for dimension alongaxis
which issplit_lengths[i]
ifsplit_lengths[i] != -1
. Otherwise, the dimension alongaxis
is processed as described insplit_lengths
input description.
Types
T1: any arbitrary supported type.
T2: any integer type.
Examples
<layer id="1" type="VariadicSplit" ...>
<input>
<port id="0"> <!-- some data -->
<dim>6</dim>
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
<port id="1"> <!-- axis: 0 -->
</port>
<port id="2">
<dim>3</dim> <!-- split_lengths: [1, 2, 3] -->
</port>
</input>
<output>
<port id="3">
<dim>1</dim>
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
<port id="4">
<dim>2</dim>
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
<port id="5">
<dim>3</dim>
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
</output>
</layer>
<layer id="1" type="VariadicSplit" ...>
<input>
<port id="0"> <!-- some data -->
<dim>6</dim>
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
<port id="1"> <!-- axis: 0 -->
</port>
<port id="2">
<dim>2</dim> <!-- split_lengths: [-1, 2] -->
</port>
</input>
<output>
<port id="3">
<dim>4</dim> <!-- 4 = 6 - 2 -->
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
<port id="4">
<dim>2</dim>
<dim>12</dim>
<dim>10</dim>
<dim>24</dim>
</port>
</output>
</layer>