pytorch 给定概率分布的张量，如何利用这个概率进行重复\不重复采样？