pub fn scatter_add_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-add updates to the array at the given indices along a single axis.
Adds the update values to the existing values at the specified indices.
ยงParams
a: Input arrayindices: Indices array specifying positions to scatter intoupdates: Values to addaxis: The axis along which to scatter