Hi
For each sample within a batch of Mnist data,  I need to select one or more samples with the same label.
is there any way to do it without going through the batch with a For loop? I mean a vectorized version
which can use GPU acceleration.