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
最新推荐文章于 2024-09-25 10:36:36 发布