Function put_along_axis 

pub fn put_along_axis(
    a: &Array,
    indices: &Array,
    values: impl Into<ScalarOrArray>,
    axis: impl Axis,
) -> Array

A copy with values placed at indices along axis; see Array::put_along_axis.