lookup.py 文件源码

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

项目:classify 作者: kupospelov 项目源码 文件源码
def get_embedding_graph(self):
        data = tf.placeholder(tf.int32, shape=[None, None], name='data')
        embeddings = tf.constant(
                self.indexer.vectors, tf.float32, name='embeddings')

        vectors = tf.map_fn(
                lambda d: tf.nn.embedding_lookup(embeddings, d),
                data,
                tf.float32)

        padded = tf.pad(
                vectors,
                [[0, 0], [0, self.max_length - tf.shape(vectors)[1]], [0, 0]])

        return {
            'padded': padded,
            'data': data
        }
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号