def neg_samples(self, batch_size: int):
n_samples = batch_size * self.n_neg_samples
return multinomial(self.output_dist, num_samples=n_samples, replacement=True)
微信
公众号
扫码关注公众号