in order to use gather i need to have x and ids of the same shape except on dim.
in my new case the shapes are
x - (5,10,15,20)
ids - (5,10)
I can use unsqueeze to make shape of ids either (5,10,1) or (5,10,1,1) but it still wont match the shape of x.