嵌入层 tf.keras.layers.Embedding() 介绍
发布日期:2021-07-01 04:20:51
浏览次数:2
分类:技术文章
本文共 869 字,大约阅读时间需要 2 分钟。
函数原型
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)
官网地址:https://tensorflow.google.cn/api_docs/python/tf/keras/layers/Embedding
作用
将正整数(索引)转换为固定大小的向量,例如[[4], [20]] -> [[0.25, 0.1], [0.6, -0.2]]
。
- Embedding层只能用作模型中的第一层
参数
- input_dim:大或等于0的整数,字典长度,即输入数据最大下标+1
- output_dim:大于0的整数,代表全连接嵌入的维度
- embeddings_initializer: 嵌入矩阵的初始化方法,为预定义初始化方法名的字符串,或用于初始化权重的初始化器。
- embeddings_regularizer: 嵌入矩阵的正则项,为
Regularizer
对象 - embeddings_constraint: 嵌入矩阵的约束项,为
Constraints
对象 - mask_zero:布尔值,确定是否将输入中的‘0’看作是应该被忽略的“填充”(
padding
)值,该参数在使用递归层处理变长输入时有用。设置为True的话,模型中后续的层必须都支持masking,否则会抛出异常。如果该值为True
,则下标0在字典中不可用,input_dim
应设置为|vocabulary| + 1
。 - input_length:当输入序列的长度固定时,该值为其长度。如果要在该层后接
Flatten
层,然后接Dense
层,则必须指定该参数,否则Dense
层的输出维度无法自动推断。
转载地址:https://mtyjkh.blog.csdn.net/article/details/115796173 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
做的很好,不错不错
[***.243.131.199]2024年04月28日 17时12分12秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
Git 常用命令笔记
2019-05-03
Springboot使用详解
2019-05-03
SHA-256 算法-java实现
2019-05-03
HTTPS配置说明文档(tomcat)
2019-05-03
关于java的模板方法设计模式
2019-05-03
java并发编程(十四)- 显示锁
2019-05-03
java并发编程(十三)- 显示锁使用Lock和Condition实现等待通知模式
2019-05-03
java并发编程(十五)-LockSupport工具类
2019-05-03
Spring源码分析(七) - bean的生命周期
2019-05-03
leetcode算法 111. 二叉树的最小深度
2019-05-03
leetcode算法 8. 字符串转换整数 (atoi)
2019-05-03
剑指offer29:顺时针打印矩阵
2019-05-03
剑指offer30:包含min函数的栈
2019-05-03
剑指offer31:栈的压入、弹出序列
2019-05-03
JVM类加载运行内存过程
2019-05-03
李洪强iOS开发之-cocopods安装
2019-05-03
iOS开发UI篇—使用UItableview完成一个简单的QQ好友列表(二)
2019-05-03
iOS开发UI篇—在UItableview中实现加载更多功能
2019-05-03