mindspore.ops.gather_nd ======================= .. py:function:: mindspore.ops.gather_nd(input_x, indices) 根据索引获取输入Tensor指定位置上的元素。 若 `indices` 是K维整型Tensor,则可看作是从 `input_x` 中取K-1维Tensor,每个元素都是一个切片: .. math:: output[(i_0, ..., i_{K-2})] = input\_x[indices[(i_0, ..., i_{K-2})]] `indices` 的最后一维的长度不能超过 `input_x` 的秩: :math:`indices.shape[-1] <= input\_x.rank` 。 **参数:** - **input_x** (Tensor) - GatherNd的输入。任意维度的Tensor。 - **indices** (Tensor) - 索引Tensor,其数据类型为int32或int64。 **返回:** Tensor,数据类型与 `input_x` 相同,shape为 `indices_shape[:-1] + input_x_shape[indices_shape[-1]:]` 。 **异常:** - **ValueError** - `input_x` 的shape长度小于 `indices` 的最后一维的长度。