Split
功能
将输入切分成多个输出。
输入
一个输入:
x:一个tensor,数据类型:float16、float、int8、int16、int32、int64、uint8、uint16、uint32、uint64。
输出
一个输出:
y:由多个输出tensor组成的列表,每个tensor数据类型和输入x一致。
属性
split:list,数据类型:int8、int16、int32、int64,指定每个输出tensor沿着切分方向的大小(Opset 13之后作为输入)。
axis:数据类型:int8、int16、int32、int64,指定切分的方向。
约束
split的每个元素必须>=1。
split的所有元素之和必须等于axis指定的切分方向的size。
axis在[ -len(x.shape), len(x.shape)-1] 之间。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单