Split

Description

Splits an input into multiple outputs.

Input

One input:

x: tensor of type float16, float, int8, int16, int32, int64, uint8, uint16, uint32 or uint64.

Output

One output:

y: list of multiple output tensors. The data type of each tensor is the same as that of input x.

Attribute

split: list of type int8, int16, int32, or int64, specifying the size of each output tensor along the split direction (after Opset13 as the input).

axis: int8, int16, int32, int64, split direction.

Constraints

Each element of split must be greater than or equal to 1.

The sum of all split elements must be equal to the size of the split direction specified by axis.

axis is between [ -len(x.shape), len(x.shape)-1].

ONNX Opset Support

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18