gather_single

mlx_rs::ops::indexing

Function gather_single 

Source 
pub fn gather_single(
    a: impl AsRef<Array>,
    indices: impl AsRef<Array>,
    axis: i32,
    slice_sizes: &[i32],
) -> Result<Array>
Expand description

Gather elements from the array at the given indices along a single axis.

ยงParams

  • a: Input array
  • indices: Indices array specifying positions to gather from
  • axis: The axis along which to gather
  • slice_sizes: The sizes of the slices to gather