pytorch learning

torch.multinomial(input,num_samples,replacement=False, out=None)

input 是一个张量，每一行的元素为非负，且每一行的元素和为正，固定的每一行中的列元素的相对大小决定了在抽取时被取到的概率大小。num_samples 表示对每一行的抽取的次数，如果replacement 为 true，则允许在每一行重复抽取，这种情况下，那一行对应的列元素为零的概率为零，就不能被抽到了。