Split

功能

将输入Tensor沿指定轴拆分成多个Tensor。

输入

input：输入Tensor，数据类型支持float16、float、int8、int16、int32、int64、uint8、uint16、uint32、uint64。

属性

split（Opset13之后作为输入）：int列表，用于指定沿拆分轴每个输出Tensor的大小，数据类型支持int8、int16、int32、int64。

axis：int，用于指定拆分的轴，数据类型支持int8、int16、int32、int64。

输出

outputs：多个输出Tensor，每个Tensor数据类型与输入input一致。

限制与约束

split：列表中的每个元素须大于1，且所有元素的和必须等于拆分维度的大小。

支持的ONNX版本

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