I have two tensors of size [b, c, h, w, dim]; one contains values, while other contains indices that these values should map to. I would like to create a third tensor of this dimension and perform the map: out = value[idx]
How can I do this? I believe index operations like index_copy don’t work with dimensions greater than 1.