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

GatherElements

功能

获取索引位置的元素产生输出。

输入

两个输入:

input:秩大于1的tensor输入,类型:float16、float、int32。

indices:int32、int64的索引tensor。

输出

一个输出:

output:一个tensor,与indices的shape相同。

属性

axis:int,默认为0 指定聚集的轴。

支持的ONNX版本

Opset v11/v12/v13/v14/v15/v16/v17/v18

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

当前产品无相关内容

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