GatherND

功能

将输入数据切片输出。

输入

两个输入：

data：秩r>=1的tensor输入，类型：float16、float、double、int32、int64等。

indices：int64的索引tensor，秩q>=1。

输出

一个输出：

output：一个tensor，秩为q + r - indices_shape[-1] - 1。

属性

batch_dims：int，批处理轴的数量，当前仅支持配置为0。

约束

不支持atc工具参数--precision_mode=must_keep_origin_dtype时double的输入。

支持的ONNX版本

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