Split

Description

Splits an input tensor into multiple tensors along a specified axis.

Input

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

Attribute

split (used as input in Opset v13 and later): int list, specifying the size of each output tensor along the split axis. The data type can be int8, int16, int32 or int64.

axis: int, specifying the axis to be split. The data type can be int8, int16, int32 or int64.

Output

outputs: multiple output tensors. Each tensor has the identical data type as the input.

Constraints

split: Each element in the list must be greater than 1, and the sum of all elements must be equal to the size of the split dimension.

ONNX Opset Support

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