從一個(gè)樣本空間中既穆,抽樣。
比如:
我要抽 [0, 1, 2] 三個(gè)物體,共100次,那我希望:
- 0能抽到20次左右
- 1能抽到70次左右
- 2能抽到30次左右
x = torch.Tensor([2, 7, 3]) #20次弄捕,70次,30次
m = torch.distributions.Categorical(x)
re = [0, 0, 0] #三個(gè)數(shù)抽到的個(gè)數(shù)
for i in range(100):
re[m.sample()] += 1 #sample就是抽一次
plt.bar([0, 1, 2], re)