功能:根据input_ids中的id,寻找embeddings中的第id行,即选取一个张量里面索引对应的元素。embedding_lookup不是简单的查表,id对应的向量是可以训练的,训练参数个数应该是 category num*embedding size,也就是说lookup是一种全连接层。
tf.nn.embedding_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的值 |
例如