torch.multinomial

torch.multinomial函数是PyTorch中用于从多项式分布中抽取样本的函数。它可以根据给定的概率分布从每个类别中抽取指定数量的样本。[1]

举个例子，假设有一个包含五个类别的多项式分布，并想从中抽取三个样本，可以使用torch.multinomial函数。首先，需要定义一个包含五个类别的概率分布，例如[0.1, 0.2, 0.3, 0.25, 0.15]。然后，将概率分布转化为二维形式，即将其reshape为(1, 5)的形状。最后，使用torch.multinomial函数抽取三个样本。[2]

需要注意的是，使用torch.multinomial函数时，输入的tensor必须满足以下要求：数据类型必须是浮点型（float或double），因为多项式分布的参数通常是实数。维度必须是二维的，其中第一维表示每个样本的概率分布，第二维表示抽取的样本数。例如，如果要从五个类别中抽取三个样本，则tensor的形状应为(5, 3)。[3]