The NWayOneShotEvalSet is the function for evaluating the model against n-way one shot. For example, if n = 4, the set outputs a batch of 4 images, with 1 of them being from the correct label. The getitem function randomly creates these sets, so the idx is not used. The iteration is set by the testSize variable. Hope this helps!