Hi, thanks for replying, but in my case idx_x and idx_y have to be 2x4 instead of just 4, meaning each example in the batch has its own indices. Is it still possible with advanced indexing? Or do we have to use a for-loop to loop through the examples in a batch?