mindspore.ops.Gather

class mindspore.ops.Gather(batch_dims=0)[source]
prim = ops.Gather(batch_dims)
out = prim(input_params, input_indices, axis)

is equivalent to

ops.gather(input_params, input_indices, axis, batch_dims)

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