Module gather 

Structs

GpuGather

Type Aliases

DispatchGatherFn

output = data.gather(axis, indices), i.e. output[..., i, ...] = data[..., indices[i], ...] along axis. Negative indices wrap (matches the CPU op).