mindspore.ops.GatherNd

class mindspore.ops.GatherNd[源代码]
prim = ops.GatherNd()
out = prim(input_x, indices)

is equivalent to

ops.gather_nd(input_x, indices)

Refer to mindspore.ops.gather_nd() for more details.