Paddle fluid.layers.embedding 如何多个 feature 间共享?

lc8prwob  于 2021-11-29  发布在  Java
关注(0)|答案(4)|浏览(307)

请问 paddle 如何实现在多 feature 间共享一个 embedding table?

qij5mzcb

qij5mzcb1#

Paddle/python/paddle/fluid/layers/nn.py

Line 357 in f8ecc3d

| | param_attr=None, |

通过param_attr 将所有embedding的参数名设置成一样 就可以共享了

voase2hg

voase2hg2#

好的,谢谢,请问 paddle embedding 是否支持变长 ids 的查找和聚合,类似 tf 的 https://www.tensorflow.org/api_docs/python/tf/nn/embedding_lookup_sparse

看官方文档示例,ids 的最后一个维度必须是 1:
http://paddlepaddle.org/documentation/docs/zh/1.4/api_cn/layers_cn.html#embedding

bzzcjhmw

bzzcjhmw3#

@lzha106 paddle fluid的embedding支持稀疏更新,但可能与tf不一样的是,参数本身不能是sparse tensor。目前确实需要ids的最后一个维度为1

iklwldmw

iklwldmw4#

@kuke ,多谢,如果是这样的话,对于变长的 feature id list,不知道是否有示例参考下 paddle 如何实现查找及聚合操作?

相关问题