pytorch系列:torch.multinomial

torch.multinomial(input, num_samples, replacement=False, out=None)返回一个每一行包含 n u m _ s a m p l e s num\_samples num_samples个从位于对应tensor的行通过多项式概率分布采样来值。

注意: i n p u t input input的行和不必是加和为1(在那些我们使用这些值作为权重的情况中),大必须是非负的,有限的和非零和。

根据每一个采样值的先后,从左到右的索引是有序的(第一个采样值放第一列)。

如果 i n p u t input input是一个向量,那么 o u t out out是一个大小为 n u m _ s a m p l e s num\_samples num_samples的向量。
如果 i n p u t input input是一个具有 m m m行的矩阵,那么 o u t out out是一个具有shape为 ( m × num_samples ) (m \times \text{num\_samples}) (m×num_samples)的矩阵。

你可能感兴趣的:(NN框架,pytorch系列)