gather_nd

该OP是 gather 的高维推广，并且支持多轴同时索引。 index 是一个K维度的张量，它可以认为是从 x 中取K-1维张量，每一个元素是一个切片：

示例：

参数：

• index (Tensor) - 输入的索引Tensor，其数据类型int32或者int64。它的维度 必须大于1，并且 index.shape[-1] <= x.rank 。

• name （str，可选）- 具体用法请参见 ，一般无需设置，默认值为None。

代码示例：

import numpy as np
np_index = [[0, 1]]
x = paddle.to_tensor(np_x)
index = paddle.to_tensor(np_index)