欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

TF-tf.keras.layers.Embedding

程序员文章站 2024-03-15 11:44:11
...
tf.keras.layers.Embedding(
    input_dim, output_dim, embeddings_initializer='uniform',
    embeddings_regularizer=None, activity_regularizer=None,
    embeddings_constraint=None, mask_zero=False, input_length=None, **kwargs
)

作用是:将正整数(索引)转换为大小固定的向量
这个层只能作为模型的第一层使用

参数:
input_dim:词汇表的大小,即最大整数索引+ 1
output_dim:密嵌入维数
embeddings_initializer:嵌入矩阵的初始化
embeddings_regularizer:将正则化函数应用于嵌入矩阵
embeddings_constraint:约束函数应用于嵌入矩阵
mask_zero:输入值0是否为应屏蔽的特殊“填充”值
input_length:输入序列的长度(恒定时)。
如果要在下面连接Flatten然后连接Dense层,则需要此参数(如果没有,则无法计算密集输出的形状)

TF-tf.keras.layers.Embedding

TF-tf.keras.layers.Embedding

相关标签: TF Embedding