pytorch,numpy根据权重取值 -- np.random.multinomial,np.random.choice和torch.multinomial函数
np.random.multinomial(n,pvals,size=None)->取到的次数分布数组该函数表示根据一个概率数组,取若干次,得到一个次数分布数组参数说明:–n:从矩阵中取值次数;–pvals:根据概率取值,这是一个数组,并且所有数据之和为1;–size:输出的维度,默认为1,即1xpvals例子:a=np.random.rand(6)print(a)#array([0.028634