功能：根据input_ids中的id，寻找embeddings中的第id行，即选取一个张量里面索引对应的元素。embedding_lookup不是简单的查表，id对应的向量是可以训练的，训练参数个数应该是 category num*embedding size，也就是说lookup是一种全连接层。
bedding_lookup(params,ids,partition_strategy='mod',name=None,validate_indices=True,max_norm=None
)
| 参数 | 必选 | 类型 | 说明 |
|---|---|---|---|
| params | 是 | 张量或n个张量的列表 | 输入的服从[0,1]的均匀分布或者标准分布的张量 |
| ids | 是 | int32或int64 | 张量索引，包含要在params中查找的id |
| partition_strategy | 否 | string | 指定分区策略的字符串，支持div默认mod |
| name | 否 | string | 运算名称 |
| validate_indices | 否 | bool | 是否验证收集索引 |
| max_norm | 否 | 如果不是None，嵌入值将被l2归一化为max_norm的值 |
例如
本文发布于:2024-01-28 23:21:28，感谢您对本站的认可！
本文链接：https://www.4u4v.net/it/170645529311031.html
版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系，我们将在24小时内删除。
| 留言与评论（共有 0 条评论） |
