如何把embedding层的向量输出?
收藏
每次训练都会建立一个embedding层,一些学习任务需要关注这层训练出来的信息,比如词向量,如何将embedding层的向量信息输出?
2
收藏
请登录后评论
我最近琢磨,请问解决了吗?
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())
谢谢三楼,帮大忙了