Solved by ptrblck in post #2 
              
              
                You could duplicate the single elements in your index tensor, so that you could create a tensor to use in .scatter_. I’m not sure, if there is a better way besides the obvious one using a loop.