如何把embedding层的向量输出?
wxm00236 发布于2019-08-14 13:09 浏览:396 回复:3
1
收藏

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

收藏
点赞
1
个赞
共3条回复 最后由130******57回复于2019-11-21 15:09
#4130******57回复于2019-11-21 15:09:56

谢谢三楼,帮大忙了

0
#31083780381回复于2019-09-19 16:53:36

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())

1
#2jnwusanshi回复于2019-09-08 10:53:22

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

0
TOP
切换版块