aa5b6ecac2
* ops to rst * fix errors * formula fix * change code * console directive * vsplit try hoghlight * fix code snippets * comment fixes * fix list
115 lines
4.1 KiB
Markdown
115 lines
4.1 KiB
Markdown
# VariadicSplit {#openvino_docs_ops_movement_VariadicSplit_1}
|
|
|
|
@sphinxdirective
|
|
|
|
**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]``.
|
|
|
|
.. math::
|
|
|
|
shape\_output\_tensor = [data.shape[0], data.shape[1], \dotsc , split\_lengths[i], \dotsc , data.shape[D-1]]
|
|
|
|
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 along ``data`` to split. A scalar or tensor with shape ``[1]`` of type ``T2`` 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 split ``axis``. A 1D tensor of type ``T2``. The number of elements in ``split_lengths`` determines the number of outputs. The sum of elements in ``split_lengths`` must match ``data.shape[axis]``. In addition ``split_lengths`` can contain a single ``-1`` element, which means, all remaining items along specified ``axis`` 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 along ``axis`` which is ``split_lengths[i]`` if ``split_lengths[i] != -1``. Otherwise, the dimension along ``axis`` is processed as described in ``split_lengths`` input description.
|
|
|
|
**Types**
|
|
|
|
* *T1*: any arbitrary supported type.
|
|
* *T2*: any integer type.
|
|
|
|
**Examples**
|
|
|
|
.. code-block:: cpp
|
|
|
|
<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>
|
|
|
|
|
|
.. code-block:: cpp
|
|
|
|
<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>
|
|
|
|
|
|
@endsphinxdirective
|
|
|