首页 Paddle框架 帖子详情
如何把embedding层的向量输出?
收藏
快速回复
Paddle框架 问答深度学习 2535 3
如何把embedding层的向量输出?
收藏
快速回复
Paddle框架 问答深度学习 2535 3

每次训练都会建立一个embedding层,一些学习任务需要关注这层训练出来的信息,比如词向量,如何将embedding层的向量信息输出?

2
收藏
回复
全部评论(3)
时间顺序
j
jnwusanshi
#2 回复于2019-09

我最近琢磨,请问解决了吗?

0
回复
1
1083780381
#3 回复于2019-09

https://www.paddlepaddle.org.cn/documentation/docs/zh/1.5/faq/train_cn.html

通过这种方式可以获取保存:

embed_first = fluid.layers.embedding(input=words[0],size=[dict_size, EMBED_SIZE],dtype='float32',is_sparse=is_sparse,
param_attr=fluid.ParamAttr(name='shared_w'))

.......

emb = numpy.array(fluid.global_scope().find_var("shared_w").get_tensor())

2
回复
SPiCa
#4 回复于2019-11

谢谢三楼,帮大忙了

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户