tensorflow中tf.multinomial函数的使用方法讲解
·
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的输出中比较多,
新一代开源开发者平台 GitCode,通过集成代码托管服务、代码仓库以及可信赖的开源组件库,让开发者可以在云端进行代码托管和开发。旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。
更多推荐


所有评论(0)