masked_scatter_device

mlx_rs::ops::indexing

Function masked_scatter_device 

Source 
pub fn masked_scatter_device(
    a: impl AsRef<Array>,
    mask: impl AsRef<Array>,
    src: impl AsRef<Array>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

Scatter values into an array at locations where mask is true.

ยงParams

  • a: Input array
  • mask: Boolean mask array indicating where to scatter
  • src: Source values to scatter