Is there a way to implement 3-D scatter_/scatter_add_ in pytorch?
Specifically, I have a tensor X (shape (B, M, N)), an index tensor I of shape (B, C, D) giving indices into the X, and a values tensor V the same shape of as indices, (B, C, D). For indices (b, c, d) in I, I want to add V[b,c,d] to X[b,c,d].
Is there a way to implement this? The regular scatter_ function only indices using a single dim.