# gather(indices:axis:batchDimensionCount:) Adds a gather operation to the current graph. ``` func gather(indices: some BNNSGraph.Builder.OperationParameter, axis: Int, batchDimensionCount: Int) -> BNNSGraph.Builder.Tensor ``` ## Parameters `indices` The indices that the operation gathers. `axis` The axis over which the operation gathers. `batchDimensionCount` The number of additional leading dimensions. ## Discussion Register a gather operation, which gathers entire slices of the input tensor along the specified axis. In particular, for each coordinate i valid for the indices tensor, fetch an input slice from the input tensor by indexing the axis dimension at `indices[i...]`, then store the entire slice into the output tensor by indexing it at `i...`. Outputs `y` such that `y[L..., i..., T...] = x[L..., indices[i...], T...]`, where `L` denotes leading dimensions before the axis, `T` denotes trailing dimensions after the axis, and i… denotes indices valid for the indices tensor. `batchDimensionCount` specifies the number of additional leading dimensions present in both `self` and `indices` which should be only used for a batch processing loop. --- Copyright © 2026 Apple Inc. All rights reserved. | [Terms of Use](https://www.apple.com/legal/internet-services/terms/site.html) | [Privacy Policy](https://www.apple.com/privacy/privacy-policy)