GatherND

功能

将输入数据切片输出

边界

【输入】

2个输入

data:秩r>=1的tensor输入,类型:float16, float32, double, int32, int64等

indices:int64的索引张量,秩q>=1

【输出】

1个输出

output:一个张量, 秩为q + r - indices_shape[-1] - 1

【属性】

batch_dims:int,默认为0 批处理轴的数量

【约束】

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

支持的ONNX版本

Opset v11/v12/v13