def embedding_layer(x, vocab_size, embedding_size):
"""
:param x: 输入tensor
:param vocab_size: 需要embedding的value的总量
:param embedding_size: embdding后tensor的axis=-1的维度
:return: embedding后的tensor
"""
# 创建一个n * k的默认值矩阵
embedding_w = tf.get_variable('embedding_w', [vocab_size, embedding_size]
Tensorflow实现Embedding Layer
使用Tensorflow训练Embedding层
最新推荐文章于 2024-09-25 10:36:36 发布
该博客介绍了如何利用Tensorflow构建并训练Embedding Layer,强调在模型训练过程中,初始的n*k维度的embedding矩阵会随训练迭代不断优化。

最低0.47元/天 解锁文章
251

被折叠的 条评论
为什么被折叠?



