scatter_single_device

mlx_rs::ops::indexing

Function scatter_single_device 

Source 
pub fn scatter_single_device(
    a: impl AsRef<Array>,
    indices: impl AsRef<Array>,
    updates: impl AsRef<Array>,
    axis: i32,
    stream: impl AsRef<Stream>,
) -> 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