Version of embedding_lookup that avoids duplicate lookups.
tf.contrib.layers.embedding_lookup_unique(
params, ids, partition_strategy='mod', name=None
)
This can save communication in the case of repeated ids.
Same interface as embedding_lookup. Except it supports multi-dimensional ids
which allows to not reshape input/output to fit gather.
Args |
|---|
params
|
A list of tensors with the same shape and type, or a
PartitionedVariable. Shape [index, d1, d2, ...].
|
ids
|
A one-dimensional Tensor with type int32 or int64 containing the
ids to be looked up in params. Shape [ids1, ids2, ...].
|
partition_strategy
|
A string specifying the partitioning strategy, relevant
if len(params) > 1. Currently "div" and "mod" are supported. Default
is "mod".
|
name
|
A name for this operation (optional).
|
Returns |
|---|
A Tensor with the same type as the tensors in params and dimension of
[ids1, ids2, d1, d2, ...].
|
Raises |
|---|
ValueError
|
If params is empty.
|
View source on GitHub