torch.multinomial()

先看一下定义(没上官网，在外，热点官网上不去)

def multinomial(self: Tensor, num_samples: _int, replacement: _bool=False, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...

参数:

• input:我也不知道为何上面的函数参数中没有，默认第一个参数是input，是一个Tensor，里面的元素值越大，那么这个元素就越有可能被取到，取出的不是这个元素值，而是这个元素对应的下标
• num_samples:整数，表示要取多少个下标
• replacement:True表示有放回的采样，False表示不放回，取一个少一个
• generator暂时没去了解是什么，懂的评论教我一手，我只知道类型必须是torch.Generator,好像必须要CUDA才能用。。

实验

import torch
# import numpy as np
freq = torch.Tensor([0.1,0.2,0.3,0.9])
K = 2
# generate = torch.Generator
res = torch.multinomial(input = freq,num_samples=K,replacement=True)
print(res)

结果

基本上每次都是:
tensor([3, 3])
但也有别的情况