index_add_ 124 might be of help. I spent several hours looking for scatter_nd like function in pytorch, but all the links pointed to scatter_add_. The tensorflow like scatter_nd is more similar to index_add_.