Split

Versioned name : Split-1

Category : Data movement

Short description : Split operation splits an input tensor into pieces of the same length along some axis.

Detailed Description

Split operation splits a given input tensor data into chunks of the same length along a scalar axis. It produces multiple output tensors based on num_splits attribute. The i-th output tensor shape is equal to the input tensor data shape, except for dimension along axis which is data.shape[axis]/num_splits.

\[shape\_output\_tensor = [data.shape[0], data.shape[1], \dotsc , data.shape[axis]/num\_splits, \dotsc data.shape[D-1]]\]

Where D is the rank of input tensor data. The axis being split must be evenly divided by num_splits attribute.

Attributes

  • num_splits

    • Description : number of outputs into which the input tensor data will be split along axis dimension. The dimension of data shape along axis must be evenly divisible by num_splits

    • Range of values : an integer within the range [1, data.shape[axis]]

    • Type : int

    • Required : yes

Inputs

  • 1 : data. A tensor of type T and arbitrary shape. Required.

  • 2 : axis. Axis along data to split. A scalar of type T_AXIS within the range [-rank(data), rank(data) - 1]. Negative values address dimensions from the end. Required.

  • Note : The dimension of input tensor data shape along axis must be evenly divisible by num_splits attribute.

Outputs

  • Multiple outputs : Tensors of type T. The i-th output has the same shape as data input tensor except for dimension along axis which is data.shape[axis]/num_splits.

Types

  • T : any arbitrary supported type.

  • T_AXIS : any integer type.

Example

<layer id="1" type="Split" ...>
    <data num_splits="3" />
    <input>
        <port id="0">       <!-- some data -->
            <dim>6</dim>
            <dim>12</dim>
            <dim>10</dim>
            <dim>24</dim>
        </port>
        <port id="1">       <!-- axis: 1 -->
        </port>
    </input>
    <output>
        <port id="2">
            <dim>6</dim>
            <dim>4</dim>
            <dim>10</dim>
            <dim>24</dim>
        </port>
        <port id="3">
            <dim>6</dim>
            <dim>4</dim>
            <dim>10</dim>
            <dim>24</dim>
        </port>
        <port id="4">
            <dim>6</dim>
            <dim>4</dim>
            <dim>10</dim>
            <dim>24</dim>
        </port>
    </output>
</layer>