scatter_single

mlx_rs::ops::indexing

Function scatter_single 

Source 
pub fn scatter_single(
    a: impl AsRef<Array>,
    indices: impl AsRef<Array>,
    updates: impl AsRef<Array>,
    axis: i32,
) -> Result<Array>
Expand description

Scatter updates to the array at the given indices along a single axis.

ยงParams

  • a: Input array
  • indices: Indices array specifying positions to scatter into
  • updates: Values to scatter
  • axis: The axis along which to scatter