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层,则需要此参数(如果没有,则无法计算密集输出的形状)