@killeent I don’t understand. Doesn’t your example show that it works fine if both the tensor and the indices are on the GPU? This is certainly how it currently works for me (but maybe this was introduced in pytorch 0.2).