Is there any pytorch function that can help select every time equal no of times from Tensor list
say tensor([0, 1])
If i make sample(tensor[0,1],1)   20 times i should nearly 50 pct of times 0 ,50 pct of time 1