mindspore.ops.Gather

class mindspore.ops.Gather(batch_dims=0)

prim = ops.Gather(batch_dims)
out = prim(input_params, input_indices, axis)

等价于

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

更多详情请查看： mindspore.ops.gather() 。