gather(operand, start_indices, dimension_numbers, slice_sizes)¶
Wraps XLA’s Gather operator.
The semantics of gather are complicated, and its API might change in the future. For most use cases, you should prefer Numpy-style indexing (e.g., x[:, (1,4,7), …]), rather than using gather directly.
Any) – an array from which slices should be taken
Any) – the indices at which slices should be taken
GatherDimensionNumbers) – a lax.GatherDimensionNumbers object that describes how dimensions of operand, start_indices and the output relate.
- Return type
An array containing the gather output.