Pytorch不放回随机抽样

WeightedRandomSampler

不放回只能用WeightedRandomSampler，其他都是放回取样

只需将权重都设为1(or相同值)

from torch.utils.data import WeightedRandomSampler,DataLoader

a = torch.Tensor([1, 2, 3, 4, 5])
dataloader = DataLoader(dataset=a,
                    batch_size=4,  # 组织成batch大小
                    sampler=WeightedRandomSampler(torch.ones(a.size()), 20))
for i in dataloader:
     print(i)

结果 

tensor([1., 5., 1., 2.])
tensor([5., 1., 5., 2.])
tensor([1., 4., 2., 2.])
tensor([4., 5., 4., 3.])
tensor([5., 2., 2., 1.])

参数

Arguments:
        weights (sequence)   : a sequence of weights, not necessary summing up to one
        num_samples (int): number of samples to draw
        replacement (bool): if ``True``, samples are drawn with replacement.
            If not, they are drawn without replacement, which means that when a
            sample index is drawn for a row, it cannot be drawn again for that row.