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.