masked_scatter

mlx_rs::ops::indexing

Function masked_scatter 

Source 
pub fn masked_scatter(
    a: impl AsRef<Array>,
    mask: impl AsRef<Array>,
    src: impl AsRef<Array>,
) -> 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