torch.multinomial
是PyTorch中的一个函数,用于从多项分布中抽取样本。多项分布是一种描述多个可能结果的概率分布,例如抛硬币、掷骰子等。
torch.multinomial的用法如下:
torch.multinomial(input, num_samples, replacement=False, *, generator=None, out=None)
参数说明:
input: 输入张量,表示每个类别的概率分
torch.multinomial
是PyTorch中的一个函数,用于从多项分布中抽取样本。多项分布是一种描述多个可能结果的概率分布,例如抛硬币、掷骰子等。
torch.multinomial的用法如下:
torch.multinomial(input, num_samples, replacement=False, *, generator=None, out=None)
参数说明:
input: 输入张量,表示每个类别的概率分