tf.compat.v1.lookup.StaticHashTable
Обобщенная хеш-таблица, которая становится неизменяемой после инициализации.
Наследуется от: StaticHashTable
tf.compat.v1.lookup.StaticHashTable(
initializer, default_value, name=None
)
При выполнении в режиме графа, вы должны оценить тензор, возвращаемый tf.tables_initializer() перед оценкой тензора, возвращаемого методом этого класса lookup(). Пример использования в режиме графа:
keys_tensor = tf.constant([1, 2])
vals_tensor = tf.constant([3, 4])
input_tensor = tf.constant([1, 5])
table = tf.lookup.StaticHashTable(
tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor), -1)
out = table.lookup(input_tensor)
with tf.Session() as sess:
sess.run(tf.tables_initializer())
print(sess.run(out))
В режиме eager, для инициализации таблицы не требуется специального кода. Пример использования в режиме eager:
tf.enable_eager_execution()
keys_tensor = tf.constant([1, 2])
vals_tensor = tf.constant([3, 4])
input_tensor = tf.constant([1, 5])
table = tf.lookup.StaticHashTable(
tf.lookup.KeyValueTensorInitializer(keys_tensor, vals_tensor), -1)
print(table.lookup(input_tensor))
| Аргументы | |
|---|---|
initializer | Инициализатор таблицы для использования. См. HashTable ядро для поддерживаемых типов ключей и значений. |
default_value | Значение, используемое, если ключ отсутствует в таблице. |
name | Имя операции (необязательно). |
| Атрибуты | |
|---|---|
default_value | Значение по умолчанию таблицы. |
initializer | |
key_dtype | Тип данных ключа таблицы. |
name | Имя таблицы. |
resource_handle | Возвращает дескриптор ресурса, связанный с этим ресурсом. |
value_dtype | Тип данных значения таблицы. |
Методы
export
export(
name=None
)
Возвращает тензоры всех ключей и значений в таблице.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Пара тензоров, первый тензор содержит все ключи, а второй тензор содержит все значения в таблице. |
lookup
lookup(
keys, name=None
)
Ищет keys в таблице, выводит соответствующие значения.
default_value используется для ключей, отсутствующих в таблице.
| Аргументы | |
|---|---|
keys | Ключи для поиска. Может быть SparseTensor или плотным Tensor. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
SparseTensor если ключи разреженные, RaggedTensor если ключи не структурированы, иначе плотный Tensor. |
| Исключения | |
|---|---|
TypeError | когда keys или default_value не совпадают с типами данных таблицы. |
size
size(
name=None
)
Вычисляет количество элементов в этой таблице.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Скалярный тензор, содержащий количество элементов в этой таблице. |
__getitem__
__getitem__(
keys
)
Ищет keys в таблице, выводит соответствующие значения.
© 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/r2.4/api_docs/python/tf/compat/v1/lookup/StaticHashTable