torch.multinomial使用

在Word2vec中使用到了非相關數據的降採樣方法, 其中用到了torch.multinomial方法, 這裏記錄一下: multinomial(input, num_samples, replacement=False) 該方法主要有三個參數, 分別是輸入的張量, 採樣的個數, 是否有重複的數據, 採樣的時候是根據輸入張量的數值當做權重來進行抽樣的, 數值越大, 抽到的可能性越大, 越小抽到的可

>>阅读原文<<