欢迎光临我的个人主页
tf.multinomial(logits, num_samples, seed=None, name=None)
看一个使用LSTM的代码,使用了这个函数,故学习下。
从multinomial分布中采样,样本个数是num_samples,每个样本被采样的概率由logits给出
参数:
logits: 2-D Tensor with shape [batch_size, num_classes]. Each slice [i, :] represents the unnormalized log probabilities for all classes.2维量,shape是 [batch_size, num_classes],每一行都是关于种类的未归一化的对数概率
num_samples: 0-D. Number of independent samples to draw for each row slice.标量,表示采样的个数,更重要的是,它限制了返回张量中元素的范围{:0,1,2,…,num_samples-1 }
返回值:
The drawn samples of shape [batch_size, num_samples],注意元素的取值范围取决于num_samples
示例: