@ptrblck I would like reserve each equal slots of tensors to each label. Since in this case there are 10 labels and 100 tensors, I am having 10 slots for each label. For instance if there are 500 tensors then each label would have 50 slots marked as 1 and rest all marked 0.