Function scatter_add_rows 

pub fn scatter_add_rows(
    dst: &mut Tensor,
    ids: &Tensor,
    src: &Tensor,
) -> Result<()>

dst[ids[r]] += src[r], in place. Embedding-table gradient scatter.