tf.contrib.layers.embedding_lookup_unique
Версия embedding_lookup, которая избегает дублированных запросов.
tf.contrib.layers.embedding_lookup_unique(
params, ids, partition_strategy='mod', name=None
)
Это может сократить объём коммуникации в случае повторяющихся идентификаторов. Имеет тот же интерфейс, что и embedding_lookup. Кроме того, поддерживает многомерные ids , что позволяет не переформировывать входные/выходные данные для соответствия gather.
| Аргументы | |
|---|---|
params | Список тензоров с одинаковой формой и типом или PartitionedVariable. Форма [index, d1, d2, ...]. |
ids | Одномерный Tensor типа int32 или int64, содержащий идентификаторы для поиска в params. Форма [ids1, ids2, ...]. |
partition_strategy | Строка, определяющая стратегию разбиения, релевантная, если len(params) > 1. В настоящее время поддерживаются "div" и "mod". По умолчанию "mod". |
name | Имя для этой операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor того же типа, что и тензоры в params, и размерностью [ids1, ids2, d1, d2, ...]. |
| Исключения | |
|---|---|
ValueError | Если params пусто. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/layers/embedding_lookup_unique
