distributions.py 文件源码

python
阅读 26 收藏 0 点赞 0 评论 0

项目:AAE-tensorflow 作者: gitmatti 项目源码 文件源码
def sample(self, dist_info):
        prob = dist_info["prob"]
        ids = tf.multinomial(tf.log(prob + TINY), num_samples=1)[:, 0]
        onehot = tf.constant(np.eye(self.dim, dtype=np.float32))
        return tf.nn.embedding_lookup(onehot, ids)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号