pytorch 重复采样 与 非重复采样

import torch
import torch.nn.functional as F
from torch.autograd import *

a = Variable(torch.FloatTensor([[0,0,0,0,0,0,90,100]]))
b=F.softmax(a,-1)

print(b.multinomial()) # 7 或 6
print(b.multinomial(2)) # 6,7 或 7,6
print(b.multinomial(2,True)) # 7,