#random integer generator
torch.randint(low=0, high=len(train_dataset), size=(6,))