文档
注册
评分
提单
论坛
小AI

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词