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

Gather

功能

根据相应的轴从“x”中收集切片。

输入

两个输入:

x1:一个tensor,数据类型支持float16、float。

indices:一个tensor,数据类型支持int32、int64。

输出

一个输出:

y:一个tensor,数据类型和输入x1类型一致。

属性

axis:数据类型支持int,指定gather的轴,取值范围为[-r, r-1](r表示输入数据的秩)。

约束

不支持indices为负值的索引。

支持的ONNX版本

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

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

当前产品无相关内容

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