gather_nd

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

    示例：

    参数：

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

    代码示例：

    1. import paddle
    2. import paddle.fluid as fluid
    4. output = fluid.layers.gather_nd(x, index)