torch.distributions.Categorical

从一个样本空间中，抽样。
比如：
我要抽 [0, 1, 2] 三个物体，共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] #三个数抽到的个数
for i in range(100):
    re[m.sample()] += 1 #sample就是抽一次

plt.bar([0, 1, 2], re)