Gather
功能
对输入Tensor x中指定的维度dim进行数据切片。
输入
- x:输入Tensor,数据类型:float16、float、int32、int64、int8、int16、uint8、uint16、uint32、uint64、bool。
- indices:输入Tensor,数据类型:int32、int64。
属性
axis:数据类型支持int,指定gather的轴,取值范围为[-r, r-1](r表示输入数据的秩)。
输出
y:输出Tensor,数据类型和输入x类型一致。
约束
不支持indices为负值的索引。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单