tf.multinomial(logits, num_samples, seed=None, name=None)
logits 大小为[batch,n_class]
num_samples 表示采样的个数
seed 随机种子数
name表示该op的名字

这个函数就是根据logits中每个类别的概率采样,这个概率可以不用归一化后的概率,也就是每个类的概率可以大于1,不需要所有类别概率和为1,我觉得它底层是会自动给我们归一化的,输出结果为[batch,num_samples]大小
表示在每个batch_size 上采样num_samples个数,他是根据logits中的概率来采样的
这个函数用在rnn的输出中比较多,

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐